如何理解np.sum tf.reduce_sum( tf.reduce_max tf.reduce_mean)等對tensor和高維矩陣的axis選擇的操作

一個不是很簡單，但是很好理解的方法是：你的輸入矩陣的shape是(2,2,4)，那麼當axis=0時，就是在第一個dimension上進行求和，最後得到的結果的shape就是去掉第一個dimension後的shape，也就是(2,4)。具體的計算方法則是，對於c[i,j,k]，假設輸出矩陣爲s[j,k]，第一個dimension求和那麼就是 s[j,k]=∑i(c[i,j,k]) 如果axis=1

>>阅读原文<<