更新时间: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)
分享:

    相关文档

    相关产品

close