mindspore.ops.Equal
- class mindspore.ops.Equal[源代码]
逐元素比较两个输入Tensor是否相等。
更多参考详见
mindspore.ops.equal()
。- 支持平台:
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]