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