mindspore.Tensor.eq

mindspore.Tensor.eq(other)

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

第二个参数可以是一个shape可以广播成第一个参数的Number或Tensor，反之亦然。

\[\begin{split}out_{i} =\begin{cases} & \text{True, if } input_{i} = other_{i} \\ & \text{False, if } input_{i} \ne other_{i} \end{cases}\end{split}\]

说明

• self 和 other 遵循隐式类型转换规则，使数据类型保持一致。

• 另一个输入必须是一个Tensor或Scalar。

• 两个输入的shape支持广播。

参数：

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

返回：

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

异常：

• TypeError - self 和 other 都不是Tensor。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> from mindspore import Tensor
>>> # case 1: The shape of two inputs are different
>>> input = Tensor([1, 2, 3], mindspore.float32)
>>> output = input.eq(2.0)
>>> print(output)
[False True False]
>>> # case 2: The shape of two inputs are the same
>>> input = Tensor([1, 2, 3], mindspore.int32)
>>> other = Tensor([1, 2, 4], mindspore.int32)
>>> output = input.eq(other)
>>> print(output)
[ True  True False]