PyTorch torch.isnan 関数
Pytorch torch リファレンスマニュアルtorch.isnanPyTorch で NaN 値かどうかを調べるための関数です。
関数定義
torch.isnan(input)
使用例
例
import torch
import math
x = torch.tensor([1.0, math.nan, 3.0, float('nan')])
result = torch.isnan(x)
print(result)
import math
x = torch.tensor([1.0, math.nan, 3.0, float('nan')])
result = torch.isnan(x)
print(result)
出力結果は次のとおりです：
tensor([False, True, False, True])
その他の拡張