mindspore.ops.IsInf

View Source On Gitee
class mindspore.ops.IsInf[source]

Determines which elements are inf or -inf for each position.

Refer to mindspore.ops.isinf() for more details.

Inputs:

  • x (Tensor) - The input tensor.

Outputs:

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

Supported Platforms:

Ascend GPU CPU

Examples

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> is_inf = ops.IsInf()
>>> x = Tensor(np.array([np.log(-1), 1, np.log(0)]), mindspore.float32)
>>> output = is_inf(x)
>>> print(output)
[False False True]
>>> x = Tensor(2.1, mindspore.float64)
>>> output = is_inf(x)
>>> print(output)
False