mindspore.mint.isinf

mindspore.mint.isinf(input)

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

$$\begin{array}{r}ou{t}_i=\{\begin{array}{ll}& True,\text{ if }{x}_i=\text{Inf}\\ & False,\text{ if }{x}_i\ne \text{Inf}\end{array}\end{array}$$

where Inf means value is infinite.

Warning

• This is an experimental API that is subject to change.

• For Ascend, it is only supported on platforms above Atlas A2.

Parameters

input (Tensor) – The input tensor.

Returns

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

Raises

TypeError – If input is not a Tensor.

Supported Platforms:

Ascend

Examples

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