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