PyTorch torch.nan_to_num 関数
Pytorch torch リファレンスマニュアルtorch.nan_to_numPyTorch において、テンソル内の NaN 値を指定した値に置き換えるための関数です。無限大の値も置き換えることができます。
関数定義
torch.nan_to_num(input, nan=0.0, posinf=None, neginf=None, out=None)
使用例
実例
import torch
# NaN を 0 に置き換える
x = torch.tensor([1.0, float('nan'), 3.0, float('inf'), float('-inf')])
result = torch.nan_to_num(x, nan=0.0)
print(result)
# NaN を 0 に置き換える
x = torch.tensor([1.0, float('nan'), 3.0, float('inf'), float('-inf')])
result = torch.nan_to_num(x, nan=0.0)
print(result)
その他の拡張