PyTorch torch.aminmax 関数

Pytorch torch 参考手册 Pytorch torch リファレンスマニュアル

torch.aminmaxPyTorch でテンソルの最小値と最大値を返すために使用される関数です。

関数定義

torch.aminmax(input, dim=None, keepdim=False)

使用例

例

import torch

x = torch.tensor([[1, 3, 2], [4, 1, 3]])

# すべての要素の最小値と最大値を返す
mm = torch.aminmax(x)
print("最小値:", mm[0])
print("最大値:", mm[1])

# dim=0 に沿った最小値と最大値
mm0 = torch.aminmax(x, dim=0)
print("dim=0 最小値:", mm0[0])
print("dim=0 最大値:", mm0[1])

# dim=1 に沿った最小値と最大値
mm1 = torch.aminmax(x, dim=1)
print("dim=1 最小値:", mm1[0])
print("dim=1 最大値:", mm1[1])

出力結果は次のとおりです：

最小值: tensor(1)
最大值: tensor(4)
dim=0 最小值: tensor([1, 1, 2])
dim=0 最大值: tensor([4, 3, 3])
dim=1 最小值: tensor([1, 1])
dim=1 最大值: tensor([3, 4])

Pytorch torch 参考手册 Pytorch torch リファレンスマニュアル

その他の拡張