首页 > 其他 > 详细

numpy笔记—np.sum中keepdims作用

时间:2019-01-18 20:32:07      阅读:347      评论:0      收藏:0      [点我收藏+]
A = np.random.randn(4,3)
B = np.sum(A, axis = 1, keepdims = True)

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

numpy笔记—np.sum中keepdims作用

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

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!