mindspore.ops.NanToNum
- class mindspore.ops.NanToNum(nan=None, posinf=None, neginf=None)[源代码]
将输入中的 NaN 、正无穷大和负无穷大值分别替换为 nan 、 posinf 和 neginf 指定的值。
警告
这是一个实验性API,后续可能修改或删除。
更多细节请参考
mindspore.ops.nan_to_num()
。- 参数:
nan (float,可选) - 替换 NaN 的值。默认值为
None
。posinf (float,可选) - 如果是一个数字,则为替换正无穷的值。如果为
None
,则将正无穷替换为 x 类型支持的上限。默认值为None
。neginf (float,可选) - 如果是一个数字,则为替换负无穷的值。如果为
None
,则将负无穷替换为 x 类型支持的下限。默认值为None
。
- 输入:
x (Tensor) - 任意维度的输入Tensor。类型必须为float32或float16。
- 输出:
Tensor,数据shape和类型与 input 相同。
- 支持平台:
Ascend
CPU
样例:
>>> import mindspore >>> import numpy as np >>> from mindspore import Tensor, ops >>> nan_to_num = ops.NanToNum() >>> x = Tensor(np.array([float('nan'), float('inf'), -float('inf'), 3.14]), mindspore.float32) >>> output = nan_to_num(x) >>> print(output) [ 0.0000000e+00 3.4028235e+38 -3.4028235e+38 3.1400001e+00]