Python Numpy中 keepdims解释

            
              import numpy as np
a = np.array([[0,3],[1,4]])
#a.shape是（2,2）
#axis=1表示按行相加，并且保持其二维特性
print(np.sum(a, axis=1, keepdims=True))

# 按行相加，不保持其二维特性
print(np.sum(a, axis=1))

            
          

输出为
array([[3], [5]])，shape是（2,1）
array([3, 5])，shape是（1,2）