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

posted @ 2019-01-18 20:22  douzujun  阅读(1621)  评论(0编辑  收藏  举报