torch.max() と torch.argmax() の違い

torch.max() と torch.argmax() の違い

記録のために: 以前はここで混乱しやすかったのですが、
torch.max () は、指定された dim の最大値と対応する位置インデックス値を返します。
torch.argmax () は、指定された dim の最大値に対応する位置インデックス値を返します。指定された明るさ。

1. torch.maxの公式使い方チュートリアル()

2. torch.argmax()の公式使い方チュートリアル

import torch

a = torch.tensor([[0.03,0.12,0.85],
                  [0.01,0.9,0.09],
                  [0.95,0.01,0.04],
                  [0.09, 0.9, 0.01]])
print(a)
print(a.dtype)

b = torch.max(a, 1)
print('b:', b)
print("b第0维度：", b[0])
print("b第1维度：", b[1])

c = torch.argmax(a, 0)
print('c第0维度:', c)
c = torch.argmax(a, 1)
print('c第1维度:', c)

実行結果は次のとおりです。

tensor([[0.0300, 0.1200, 0.8500],
        [0.0100, 0.9000, 0.0900],
        [0.9500, 0.0100, 0.0400],
        [0.0900, 0.9000, 0.0100]])
torch.float32
b: torch.return_types.max(
values=tensor([0.8500, 0.9000, 0.9500, 0.9000]),
indices=tensor([2, 1, 0, 1]))
b第0维度： tensor([0.8500, 0.9000, 0.9500, 0.9000])
b第1维度： tensor([2, 1, 0, 1])
c第0维度: tensor([2, 1, 0])
c第1维度: tensor([2, 1, 0, 1])