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