快捷方式

torch.nan_to_num

torch.nan_to_num(input, nan=0.0, posinf=None, neginf=None, *, out=None) Tensor

nanposinfneginf 指定的值分别替换 NaN、正无穷大和负无穷大值。默认情况下,NaN 被替换为零,正无穷大被替换为 input 的数据类型可以表示的最大有限值,负无穷大被替换为 input 的数据类型可以表示的最小有限值。

参数
  • input (Tensor) – 输入张量。

  • nan (Number, optional) – 用来替换 NaN 的值。默认值为零。

  • posinf (Number, optional) – 如果是数字,则用它替换正无穷大值。如果为 None,则正无穷大值被替换为 input 数据类型可以表示的最大有限值。默认值为 None。

  • neginf (Number, optional) – 如果是数字,则用它替换负无穷大值。如果为 None,则负无穷大值被替换为 input 数据类型可以表示的最小有限值。默认值为 None。

关键字参数

out (Tensor, optional) – 输出张量。

示例

>>> x = torch.tensor([float('nan'), float('inf'), -float('inf'), 3.14])
>>> torch.nan_to_num(x)
tensor([ 0.0000e+00,  3.4028e+38, -3.4028e+38,  3.1400e+00])
>>> torch.nan_to_num(x, nan=2.0)
tensor([ 2.0000e+00,  3.4028e+38, -3.4028e+38,  3.1400e+00])
>>> torch.nan_to_num(x, nan=2.0, posinf=1.0)
tensor([ 2.0000e+00,  1.0000e+00, -3.4028e+38,  3.1400e+00])

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的答案

查看资源