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.]