mindspore.ops.FloatStatus

class mindspore.ops.FloatStatus[源代码]

确定元素是否包含非数字（NaN）、正无穷还是负无穷。0表示正常，1表示溢出。

输入：

x (Tensor) - 输入Tensor。数据类型必须为float16、float32或float64。 \((N,*)\) ，其中 \(*\) 表示任意数量的附加维度。

输出：

Tensor，shape为 \((1,)\) ，数据类型为 mindspore.dtype.float32 。

异常：

TypeError - 如果 x 的数据类型不是float16，float32或float64。

支持平台：

GPU

样例：

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> float_status = ops.FloatStatus()
>>> x = Tensor(np.array([np.log(-1), 1, np.log(0)]), mindspore.float32)
>>> result = float_status(x)
>>> print(result)
[1.]