[人工智能]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列

关于shape[0]与shape[1]
shape[0]输出矩阵的行数
shape[1]表示矩阵列数