比较与torch.eq的功能差异

torch.eq

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

更多内容详见torch.eq。

mindspore.ops.equal

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

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

差异对比

PyTorch：逐元素比较两个输入Tensor是否相等。

MindSpore：MindSpore此API实现功能与PyTorch一致，仅参数名不同。

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

代码示例1

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

# PyTorch
import torch
from torch import tensor

input1 = tensor([1, 2], dtype=torch.float32)
other = tensor([[1, 2], [0, 2], [1, 3]], dtype=torch.int64)
out = torch.eq(input1, other).numpy()
print(out)
# [[ True  True]
#  [False  True]
#  [ True False]]

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

x = Tensor(np.array([1, 2]), mindspore.float32)
y = Tensor(np.array([[1, 2], [0, 2], [1, 3]]), mindspore.int64)
output = mindspore.ops.equal(x, y)
print(output)
# [[ True  True]
#  [False  True]
#  [ True False]]

代码示例2

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

# PyTorch
import torch
from torch import tensor

input1 = tensor([1, 3, 1, 4], dtype=torch.int32)
out = torch.eq(input1, 1).numpy()
print(out)
# [ True False  True False]

# MindSpore
import mindspore
from mindspore import Tensor

x = Tensor([1, 3, 1, 4], mindspore.int32)
output = mindspore.ops.equal(x, 1)
print(output)
# [ True False  True False]