keepdims的作用

keepdims的作用

例如运行如下代码

x = np.array([
    [9, 2, 5, 0, 0],
    [7, 5, 0, 0 ,0]])
x_sum = np.sum(x_exp, axis=1,keepdims=True)

• 如果设置keepdims=True则x_sum.shape为(2,1)，也就是说缩减后的轴将作为尺寸为 1 的维数留在结果中。
• 如果设置keepdims=False或者传递默认值，则x_sum.shape为(2,)， 即缩减后的轴不会保留在结果中。如果接下来我们希望进行例如broadcast或一些操作则会报错。所以通常建议将 keepdims设为True