当前位置:主页 > 软件编程 > Python代码 >

numpy求平均值的维度设定的例子

时间:2020-12-06 09:17:54 | 栏目:Python代码 | 点击:

废话不多说,我就直接上代码吧!

>>> a = np.array([[1, 2], [3, 4]]) 
>>> np.mean(a) # 将上面二维矩阵的每个元素相加除以元素个数(求平均数) 
2.5 
>>> np.mean(a, axis=0) # axis=0,计算所有子数组的平均值 
array([ 2., 3.]) 
>>> np.mean(a, axis=1) # axis=1,对每一个子数组,计算它的平均值
array([ 1.5, 3.5]) 

使用以下代码验证正确

X = np.mean(last_layer_delta_of_all_input, axis=0)
print 'X:', X
print 'X_shape:', X.shape
print 'own mean'
X = np.zeros((10,1))
for delta in last_layer_delta_of_all_input:
 X += delta
X /= 10
print 'own X:', X

您可能感兴趣的文章:

相关文章