mindspore.ops.coo_isfinite

mindspore.ops.coo_isfinite(x: COOTensor)

Determines which elements are finite for each position.

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

Parameters

x (COOTensor) – The input COOTensor.

Returns

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

Raises

TypeError – If x is not a COOTensor.

Supported Platforms:

Ascend GPU CPU

Examples

>>> from mindspore import dtype as mstype
>>> from mindspore import Tensor, ops, COOTensor
>>> indices = Tensor([[0, 1], [1, 2]], dtype=mstype.int64)
>>> values = Tensor([-1, 2], dtype=mstype.float32)
>>> shape = (3, 4)
>>> x = COOTensor(indices, values, shape)
>>> output = ops.coo_isfinite(x)
>>> print(output.values)
[ True  True]