2020-12-14

随笔：np.mean()函数
假设数据为三维

import torch
import numpy as np

a=np.array([[[1,2,3],[2,3,4]],
            [[4,5,6],[5,6,7]]])
a.shape
#(2, 2, 3)

np.mean(a)
#4.0

np.mean(a,axis=0)#
#array([[2.5, 3.5, 4.5],
       [3.5, 4.5, 5.5]])

np.mean(a,axis=1)
#array([[1.5, 2.5, 3.5],
       [4.5, 5.5, 6.5]])

np.mean(a,axis=(0,1))
#array([3., 4., 5.])

第一次写博客，还不知道怎么把jupyter的out表示出来…
请大家指教~