numpy求多维（四维）数组均值

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,)