np.argmax和torch.max

1.arg就是argument,自变量的意思

argmax(f) 返回函数f的值取最大值时自变量的值

np.argmax(a)
取回对应的索引index(从下标0开始)

2.多个维度

1.np.argmax(）

a = np.array([[1, 5, 5, 2],
              [9, 6, 2, 8],
              [3, 7, 9, 1]])

np.argmax(a,axis=0)
结果：
array([1, 2, 2, 1], dtype=int64)

很明显，按照axis=0
就是每列最大值所对应的下标，从0开始计算的

2.np.max()

np.max(a,axis=1)

结果：
array([5, 9, 9])

3.torch.max()

返回一个元组tuple
max,index