PyTorch torch.isfinite 関数


Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

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)

出力結果は次のとおりです:

tensor([True, False, False, True])

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

その他の拡張