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。
numpy中的axis和Pytorch中的dim参数
原文：https://www.cnblogs.com/liujianing/p/13236448.html
