文档首页>
Atlas 200 DK 开发者套件(1.0.7.alpha)>
TBE自定义算子开发指南>
接口参考>
标量双目(Gather模式)>
矢量计算>
TIK API>
算子代码实现接口>
vec_adds
更新时间:2021-03-18 GMT+08:00
vec_adds
功能说明
矢量内每个element与标量求和:
函数原型
vec_adds(mask, dst, src, scalar, repeat_times, dst_rep_stride, src_rep_stride, mask_mode="normal")
参数说明
请参见参数说明。
dst/src/scalar操作数的类型需要保持一致 。
昇腾310 AI处理器,dst/src/scalar支持的数据类型为:Tensor(float16,float32)。
返回值
无
注意事项
请参见注意事项。
调用示例
from te import tik tik_instance = tik.Tik() src_ub = tik_instance.Tensor("float16", (128,), name="src_ub", scope=tik.scope_ubuf) scalar = tik_instance.Scalar(dtype="float16", init_value=2) dst_ub = tik_instance.Tensor("float16", (128,), name="dst_ub", scope=tik.scope_ubuf) tik_instance.vec_adds(128, dst_ub, src_ub, scalar, 1, 8, 8)
父主题: 标量双目(Gather模式)
