np.sum(a, axis=0)
axis=0时将矩阵同列相加
axis=1时将矩阵同行相加
np.sum(a, axis=1, keepdims=True)
keepdims是为了防止sum的输出结果变成(n,) 而需要的格式时(n,1) 如上图所示
np.sum(a, axis=0)
axis=0时将矩阵同列相加
axis=1时将矩阵同行相加
np.sum(a, axis=1, keepdims=True)
keepdims是为了防止sum的输出结果变成(n,) 而需要的格式时(n,1) 如上图所示