mindspore.ops.Equal

class mindspore.ops.Equal[源代码]

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

更多参考详见 mindspore.ops.equal()

输入:
  • x (Union[Tensor, Number]) - 第一个输入可以是数值型,也可以是数据类型为数值型的Tensor。

  • y (Union[Tensor, Number]) - 当第一个输入是Tensor时,第二个输入是数值型或数据类型为数值型的Tensor。数据类型与第一个输入相同。

输出:

Tensor,shape与输入 xy 广播后的shape相同,数据类型为bool。

支持平台:

Ascend GPU CPU

样例:

>>> # case 1: The shape of two inputs are different
>>> x = Tensor(np.array([1, 2, 3]), mindspore.float32)
>>> equal = ops.Equal()
>>> output = equal(x, 2.0)
>>> print(output)
[False True False]
>>> # case 2: The shape of two inputs are the same
>>> x = Tensor(np.array([1, 2, 3]), mindspore.int32)
>>> y = Tensor(np.array([1, 2, 4]), mindspore.int32)
>>> equal = ops.Equal()
>>> output = equal(x, y)
>>> print(output)
[ True  True False]