la_xl的博客

torch.argmax(input, dim=None, keepdim=False) 转载至 ： https://blog.csdn.net/qq_46092061/article/details/120612102 torch.argmax(input, dim=None, keepdim=False) 返回指定维度最大值的序号index 函数中dim表示该维度会消失。dim=0，表示我们不要行了，求出每一列的最大值的下标;dim=1，表示我们不要列了，保留每一行的最大值的下标。 二维例子