mindspore.mint.eq

mindspore.mint.eq(input, 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}\]

说明

  • inputother 遵循隐式类型转换规则,使数据类型保持一致。

  • 输入必须是两个Tensor,或一个Tensor和一个Scalar。

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

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

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

返回:

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

异常:
  • TypeError - inputother 都不是Tensor。

支持平台:

Ascend

样例:

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