文档
注册
评分
提单
论坛
小AI

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]
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词