关于axis

设axis=i，则沿着第i个下标变化的方向进行操作。
可以按照shape(size）来理解axis
假如size[3,4,5],
axis =0 得到4,5 (3那一维被操作）
axis = 1 得到 3,5 (4那一维被操作）
axis = 2或-1 得到3,4 (5那一维被操作）

更详细见https://blog.csdn.net/zenghaitao0128/article/details/78905593