torch.argmax中dim詳解

torch.argmax()函數 argmax函數：torch.argmax(input, dim=None, keepdim=False)返回指定維度最大值的序號，dim給定的定義是：the demention to reduce.也就是把dim這個維度的，變成這個維度的最大值的index ...