CANN/pypto按位异或操作API文档

# pypto.bitwise_xor

【免费下载链接】pyptoPyPTO（发音: pai p-t-o）：Parallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto

产品支持情况

功能说明

逐元素地将input与other值进行按位异或（XOR）运算。计算公式如下：

$$ res_i = input_i \oplus other_i $$

函数原型

bitwise_xor(input: Tensor, other: Union[Tensor, int]) -> Tensor

参数说明

返回值说明

返回输出Tensor，Tensor的数据类型和input、other相同，Shape为input和other广播后大小。

约束说明

1. input 和 other 类型应该相同。
2. other 为数字的时候，不支持隐式转化。
3. other 不支持nan、inf等特殊值
4. 由于存在临时内存使用，TileShape大小有额外约束，假设TileShape为[a,b,c,d]，那么a*b*c*d*sizeof(self) + a*b*c*d*sizeof(other) + a*b*c*d*sizeof(self) < UB。

调用示例

TileShape设置示例

调用该operation接口前，应通过set_vec_tile_shapes设置TileShape。

TileShape维度应和输出一致。

如非广播场景，输入input shape为[m, n]，other为[m, n]，输出为[m, n]，TileShape设置为[m1, n1]，则m1, n1分别用于切分m, n轴。

广播场景，输入input shape为[m, n]，other为[m, 1]，输出为[m, n]，TileShape设置为[m1, n1]，则m1, n1分别用于切分m, n轴。

pypto.set_vec_tile_shapes(4, 16)

接口调用示例

a = pypto.tensor([2], pypto.DT_INT16) b = pypto.tensor([2], pypto.DT_INT16) out = pypto.bitwise_xor(a, b)

结果示例如下：

输入数据a: [2, 5] 输入数据b: [1, 7] 输出数据out: [3, 2]

【免费下载链接】pyptoPyPTO（发音: pai p-t-o）：Parallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto