numpy中的axis和Pytorch中的dim参数

1、二维：

axis=0表示分别对每一列做运算；

axis=1表示分别对每一行做运算。

2、高维：

假设输入的形状是(m, n, k):
-如果指定axis(dim)=0, 输出的size就是(1, n, k)或者(n, k)
-如果指定axis(dim)=1, 输出的size就是(m, 1, k)或者(m, k)
-如果指定axis(dim)=2, 输出的size就是(m, n, 1)或者(m, n).
size中是否有“1”，取决于参数keepdims(或keepdim)。keepdims=True会保留维度1，通常默认是等于False。