PyTorch torch.amin 関数

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

torch.amin指定した次元に沿ってテンソルの最小値を返す、PyTorch の関数です。

関数定義

torch.amin(input, dim, keepdim=False)

使用例

例

import torch

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

# すべての要素の最小値を返す
print("全体最小:", torch.amin(x))

# dim=0 に沿った最小値
print("dim=0 最小:", torch.amin(x, dim=0))

# dim=1 に沿った最小値
print("dim=1 最小:", torch.amin(x, dim=1))

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

全局最小: tensor(1)
dim=0 最小: tensor([1, 1, 2])
dim=1 最小: tensor([1, 1])

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

その他の拡張