PyTorch torch.isfinite 関数
torch.isfinitePyTorch でテンソルの要素が有限値かどうかをチェックする関数です。各要素について、有限値(無限大または NaN ではない)の場合は True となるブールテンソルを返します。
関数定義
torch.isfinite(input)
使用例
実例
import torch
# さまざまな値を含むテンソルを作成
x = torch.tensor([1.0, float('inf'), float('nan'), 2.0])
# 有限値かどうかを確認
result = torch.isfinite(x)
print(result)
# さまざまな値を含むテンソルを作成
x = torch.tensor([1.0, float('inf'), float('nan'), 2.0])
# 有限値かどうかを確認
result = torch.isfinite(x)
print(result)
出力結果は次のとおりです:
tensor([True, False, False, True])
その他の拡張
Pytorch torch リファレンスマニュアル