numpy求均值
我们知道通过“np.mean(a)”可以求得N为矩阵左右元素的均值,但是有时候我们要求多维数组特定维度的均值我们该怎么办呢?
假如我们要求多维数组hsi_target除第二维之外的其他维度的均值。其中hsi_target的维度信息为:(1,31,1392,1300),我们要得到一个长度为31的一维数组。
已知:np.mean(a,axis=0)
为每列的均值,np.mean(a,axis=1)
为求每一行的均值,因此np.mean(a,axis=n)
为求第n维的均值,因此我们可以利用嵌套的方法求多维数组某些维度的均值。
dome
hsi_target = target.cpu().numpy()
hsi_output = output.cpu().numpy()
mean_target = np.mean(np.mean(np.mean(hsi_target,0),1),1)
mean_output = np.mean(np.mean(np.mean(hsi_output,0),1),1)
print(mean_target.shape)
print(mean_output.shape)
#输出结果为:(31,)
#(31,)