pytorch中的 dim用法 shape[0]shape[1]意义

关于dim的用法
1)dim的不同值表示不同维度。dim=0表示二维中的行,dim=1在二维矩阵中表示列。以此类推0123分别表示第0123维度。

关于torch.argmax():
torch.argmax()函数中dim表示该维度会消失。argmax表示得到最大值的序号索引。
对于一个矩阵(a,b)
当我们想要求每一行最大的列标号,我们就要指定dim=1,表示我们不要列了,保留行的size就可以了。最后得出的是(a,1)a行1列
假如我们想求每一列的最大行标,就可以指定dim=0,表示我们不要行了。最后得出(1,b)1行b列
pytorch中的 dim用法 shape[0]shape[1]意义_第1张图片
关于shape[0]与shape[1]
shape[0]输出矩阵的行数
shape[1]表示矩阵列数
pytorch中的 dim用法 shape[0]shape[1]意义_第2张图片

你可能感兴趣的:(pytorch中的 dim用法 shape[0]shape[1]意义)