vec_not
功能说明
每个element按位取反:
函数原型
vec_not(mask, dst, src, repeat_times, dst_rep_stride, src_rep_stride)
参数说明
请参见参数说明。
dst与src的数据类型需要保持一致。
Atlas 200/300/500 推理产品,dst/src支持的数据类型为:Tensor(uint16/int16)
Atlas 训练系列产品,dst/src支持的数据类型为:Tensor(uint16/int16)
Atlas推理系列产品(Ascend 310P处理器)AI Core,dst/src支持的数据类型为:Tensor(uint16/int16)
Atlas推理系列产品(Ascend 310P处理器)Vector Core,dst/src支持的数据类型为:Tensor(uint16/int16)
Atlas A2训练系列产品/Atlas 800I A2推理产品,dst与src支持的数据类型为:Tensor(uint16/int16)
Atlas 200/500 A2推理产品,dst与src支持的数据类型为:Tensor(uint16/int16)
返回值
无
支持的型号
Atlas 200/300/500 推理产品
Atlas 训练系列产品
Atlas推理系列产品(Ascend 310P处理器)AI Core
Atlas推理系列产品(Ascend 310P处理器)Vector Core
Atlas A2训练系列产品/Atlas 800I A2推理产品
Atlas 200/500 A2推理产品
注意事项
请参考注意事项。
调用示例
此样例是针对数据量较小、一次搬运就可以完成的场景,目的是让大家了解接口的功能,更复杂的数据量较大的样例可参见调用示例。
from tbe import tik tik_instance = tik.Tik() src_gm = tik_instance.Tensor("int16", (128,), name="src_gm", scope=tik.scope_gm) dst_gm = tik_instance.Tensor("int16", (128,), name="dst_gm", scope=tik.scope_gm) src_ub = tik_instance.Tensor("int16", (128,), name="src_ub", scope=tik.scope_ubuf) dst_ub = tik_instance.Tensor("int16", (128,), name="dst_ub", scope=tik.scope_ubuf) # 将用户输入数据从gm搬运到ub tik_instance.data_move(src_ub, src_gm, 0, 1, 8, 0, 0) tik_instance.vec_not(128, dst_ub, src_ub, 1, 8, 8) # 将计算结果从ub搬运到目标gm tik_instance.data_move(dst_gm, dst_ub, 0, 1, 8, 0, 0) tik_instance.BuildCCE(kernel_name="vec_not", inputs=[src_gm], outputs=[dst_gm])
结果示例:
输入数据(src_gm): [ 9 -2 8 3 5 5 -3 -3 8 -6 2 7 -4 -6 -4 -3 -9 8 6 5 -8 -8 9 0 -4 -8 -7 5 -9 7 -5 -5 4 6 7 -1 9 1 5 2 -1 -4 8 -2 6 2 -3 8 -2 -9 0 -6 0 9 0 1 2 1 1 -4 9 -3 -6 9 5 -2 6 4 -5 -7 -7 -6 6 -3 0 3 8 0 5 9 2 -7 -8 0 6 9 9 -3 -3 -7 3 8 9 -7 8 3 2 3 -8 0 9 0 6 -9 8 -9 1 -2 -5 -3 -8 0 7 -9 -1 7 -1 0 -3 1 1 9 -1 6 -9 -2 9 0] 输出数据(dst_gm): [-10 1 -9 -4 -6 -6 2 2 -9 5 -3 -8 3 5 3 2 8 -9 -7 -6 7 7 -10 -1 3 7 6 -6 8 -8 4 4 -5 -7 -8 0 -10 -2 -6 -3 0 3 -9 1 -7 -3 2 -9 1 8 -1 5 -1 -10 -1 -2 -3 -2 -2 3 -10 2 5 -10 -6 1 -7 -5 4 6 6 5 -7 2 -1 -4 -9 -1 -6 -10 -3 6 7 -1 -7 -10 -10 2 2 6 -4 -9 -10 6 -9 -4 -3 -4 7 -1 -10 -1 -7 8 -9 8 -2 1 4 2 7 -1 -8 8 0 -8 0 -1 2 -2 -2 -10 0 -7 8 1 -10 -1]
父主题: 单目