Python PyTorch nan_to_num用法及代碼示例

本文簡要介紹python語言中 torch.nan_to_num 的用法。

用法:

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

參數：

• input(Tensor) -輸入張量。

• nan(數字,可選的) -替換 NaN 的值。默認為零。

• posinf(數字,可選的) -如果是數字，則用該值替換正無窮大值。如果為 None，正無窮大值將替換為 input 的 dtype 可表示的最大有限值。默認為無。

• neginf(數字,可選的) -如果是數字，則用該值替換負無窮大值。如果為 None，則負無窮大值將替換為 input 的 dtype 可表示的最低有限值。默認為無。

關鍵字參數：

out(Tensor,可選的) -輸出張量。

將 input 中的 NaN 、正無窮大和負無窮大值分別替換為 nan 、 posinf 和 neginf 指定的值。默認情況下，NaN s 被替換為 0，正無窮大被替換為 input 的 dtype 可表示的最大有限值，負無窮大被替換為 input 的 dtype 可表示的最小有限值。

例子：

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