比较与torch.bitwise_or的功能差异

torch.bitwise_or

torch.bitwise_or(input, other, *, out=None) -> Tensor

更多内容详见torch.bitwise_or。

mindspore.ops.bitwise_or

mindspore.ops.bitwise_or(x, y) -> Tensor

更多内容详见mindspore.ops.bitwise_or。

差异对比

PyTorch：如果输入的两tensor数据类型为布尔类型则计算两tensor数据的逻辑或，否则计算两tensor数据的按位或。

MindSpore：MindSpore此API实现功能与PyTorch一致，但MindSpore不支持布尔类型的tensor数据。

分类	子类	PyTorch	MindSpore	差异
参数	参数1	input	x	功能一致，参数名不同
	参数2	other	y	功能一致，参数名不同
	参数3	out	-	不涉及

代码示例1

两API实现功能一致，用法相同。

# PyTorch
import numpy as np
import torch
from torch import tensor

input = torch.tensor(np.array([0, 0, 1, -1, 1, 1, 1]), dtype=torch.int32)
other = torch.tensor(np.array([0, 1, 1, -1, -1, 2, 3]), dtype=torch.int32)
output = torch.bitwise_or(input, other).numpy()
print(output)
# [ 0  1  1 -1 -1  3  3]

# MindSpore
import numpy as np
import mindspore
import mindspore.ops as ops
from mindspore import Tensor

x = Tensor(np.array([0, 0, 1, -1, 1, 1, 1]), mindspore.int32)
y = Tensor(np.array([0, 1, 1, -1, -1, 2, 3]), mindspore.int32)
output = ops.bitwise_or(x, y)
print(output)
# [ 0  1  1 -1 -1  3  3]