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