1. 程式人生 > >numpy筆記—np.sum中keepdims作用

numpy筆記—np.sum中keepdims作用

font randn 們的 and 減少 true dom 代碼 shape

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作用