mindspore.ops.Equal
- class mindspore.ops.Equal[源代码]
逐元素比较两个输入Tensor是否相等。
更多参考详见
mindspore.ops.equal()
。- 输入:
x (Union[Tensor, Number]) - 第一个输入可以是数值型,也可以是数据类型为数值型的Tensor。
y (Union[Tensor, Number]) - 当第一个输入是Tensor时,第二个输入是数值型或数据类型为数值型的Tensor。数据类型与第一个输入相同。
- 输出:
Tensor,shape与输入 x 和 y 广播后的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]