numpy笔记—np.sum中keepdims作用

A = np.random.randn(4,3)
B = np.sum(A, axis = 1, keepdims = True)

我们使用(keepdims = True)来确保 A.shape 是(4,1)而不是(4,),它使我们的代码更加严格。容易减少深度学习中代码bug

原文地址:https://www.cnblogs.com/douzujun/p/10289560.html