import numpy as np?
X = np.array([[1,2], [4,5], [7,8]])?
print(np.mean(X, axis=0, keepdims=True))?
print(np.mean(X, axis=1, keepdims=True))
axis=0，那么輸出矩陣是1行贫途，求每一列的平均（按照每一行去求平均）薛躬；axis=1莱预，輸出矩陣是1列，求每一行的平均（按照每一列去求平均）愕宋。還可以這么理解粪薛，axis是幾，那就表明哪一維度被壓縮成1蛾扇。
再舉個(gè)更復(fù)雜點(diǎn)的例子，比如我們輸入為batch = [128, 28, 28]魏滚，可以理解為batch=128镀首，圖片大小為28×28像素，我們相求這128個(gè)圖片的均值鼠次，應(yīng)該這么寫
m = np.mean(batch, axis=0)
輸出結(jié)果m的shape為(28,28)更哄，就是這128個(gè)圖片在每一個(gè)像素點(diǎn)平均值。
