mindspore.ops.eq
- mindspore.ops.eq(input, other)[source]
Computes the equivalence between two tensors element-wise.
The second argument can be a number or a tensor whose shape is broadcastable with the first argument and vise versa.
Note
input and other comply with the implicit type conversion rules to make the data types consistent.
The input must be two Tensors, or a Tensor and a Scalar.
The shapes of the inputs can be broadcasted to each other.
- Parameters
input (Union[Tensor, Number]) – The first input is a number or a tensor whose data type is number.
other (Union[Tensor, Number]) – The second input is a number when the first input is a tensor. The data type is the same as the first input. If the first input is a number, the second input should be a tensor.
- Returns
Tensor, the shape is the same as the one after broadcasting, and the data type is bool.
- Raises
TypeError – If neither input nor other is a Tensor.
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> import mindspore >>> from mindspore import Tensor, ops >>> # case 1: The shape of two inputs are different >>> x = Tensor([1, 2, 3], mindspore.float32) >>> output = ops.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 = ops.eq(x, y) >>> print(output) [ True True False]