torch.nan_to_num¶
- torch.nan_to_num(input, nan=0.0, posinf=None, neginf=None, *, out=None) Tensor ¶
用
nan
、posinf
和neginf
指定的值分别替换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])