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]