若一个矩阵A的shape为[2,5,4]
axis=0 将2去掉,取A的[:,5,4]
axis=1 将5去掉,取A的[2,:,4]
axis=2 将4去掉,取A的[2,5,:]
axis=[1,2] 将2,5去掉,取A的[:,:,4]
在numpy与torch的求和函数,mean函数中经常用到axis与keepdims参数
keepdims(保持维度)参数解释:
1.keepdims=True; 将aixs中的维度改成1
ex:axis=0,keepdims=True,A取:[1,5,4]
在numpy/torch的求和函中举例说明:
在使用torch的和函数时,参数axis和Keepdims参数一起会报错,将axis改为dim就可以了