PyTorch torch.any 関数
Pytorch torch リファレンスマニュアルtorch.anyは PyTorch において、要素が True かどうかを判定する関数です。
関数定義
torch.any(input, dim, keepdim, out)
使用例
例
import torch
x = torch.tensor([[True, False, False], [False, False, False]])
print("True があります:", torch.any(x))
print("dim=0 に True があります:", torch.any(x, dim=0))
x = torch.tensor([[True, False, False], [False, False, False]])
print("True があります:", torch.any(x))
print("dim=0 に True があります:", torch.any(x, dim=0))
出力結果は次のとおりです。
有 True: tensor(True) dim=0 有 True: tensor([True, False, False])
その他の拡張