更新时间:2021-03-18 GMT+08:00
分享

vec_max

功能说明

按element求最大值:

函数原型

vec_max(mask, dst, src0, src1, repeat_times, dst_rep_stride, src0_rep_stride, src1_rep_stride)

参数说明

请参见参数说明

dst/src0/src1的数据类型需要保持一致。

昇腾310 AI处理器,dst/src0/src1支持的数据类型为:Tensor(float16,float32,int32)

返回值

注意事项

请参考注意事项

调用示例

from te import tik
tik_instance = tik.Tik()
src0_ub = tik_instance.Tensor("float32", (64,), name="src0_ub", scope=tik.scope_ubuf)
src1_ub = tik_instance.Tensor("float32", (64,), name="src1_ub", scope=tik.scope_ubuf)
dst_ub = tik_instance.Tensor("float32", (64,), name="dst_ub", scope=tik.scope_ubuf)
tik_instance.vec_max(64, dst_ub, src0_ub, src1_ub, 1, 8, 8, 8)
分享:

    相关文档

    相关产品

关闭导读