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