mindspore.ops.IsNan

class mindspore.ops.IsNan(*args, **kwargs)[source]

Determines which elements are NaN for each position.

Inputs:
  • input_x (Tensor) - The input tensor.

Outputs:

Tensor, has the same shape of input, and the dtype is bool.

Raises

TypeError – If input_x is not a Tensor.

Supported Platforms:

GPU

Examples

>>> is_nan = ops.IsNan()
>>> input_x = Tensor(np.array([np.log(-1), 1, np.log(0)]), mindspore.float32)
>>> output = is_nan(input_x)
>>> print(output)
[True False False]