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
原创
2020-12-14 16:18:55 ·
56 阅读 ·
0 评论