PyTorch torch.amax 関数
Pytorch torch リファレンスマニュアルtorch.amaxPyTorch で、指定された次元に沿ったテンソルの最大値を返す関数です。
関数定義
torch.amax(input, dim, keepdim=False)
使用例
例
import torch
x = torch.tensor([[1, 3, 2], [4, 1, 3]])
# すべての要素の最大値を返す
print("全体の最大:", torch.amax(x))
# dim=0 に沿った最大値
print("dim=0 最大:", torch.amax(x, dim=0))
# dim=1 に沿った最大値
print("dim=1 最大:", torch.amax(x, dim=1))
x = torch.tensor([[1, 3, 2], [4, 1, 3]])
# すべての要素の最大値を返す
print("全体の最大:", torch.amax(x))
# dim=0 に沿った最大値
print("dim=0 最大:", torch.amax(x, dim=0))
# dim=1 に沿った最大値
print("dim=1 最大:", torch.amax(x, dim=1))
出力結果は次のとおりです：
全局最大: tensor(4) dim=0 最大: tensor([4, 3, 3]) dim=1 最大: tensor([3, 4])
その他の拡張