PyTorch torch.argmin 関数
Pytorch torch リファレンスマニュアルtorch.argminPyTorch で次元に沿った最小値のインデックスを返すための関数です。
関数定義
torch.argmin(input, dim, keepdim)
使用例
実例
import torch
x = torch.tensor([[1, 3, 2], [4, 1, 3]])
print("グローバル最小インデックス:", torch.argmin(x))
print("dim=1 最小インデックス:", torch.argmin(x, dim=1))
x = torch.tensor([[1, 3, 2], [4, 1, 3]])
print("グローバル最小インデックス:", torch.argmin(x))
print("dim=1 最小インデックス:", torch.argmin(x, dim=1))
出力結果は次のとおりです：
全局最小索引: tensor(0) dim=1 最小索引: tensor([0, 1])
その他の拡張