给定一个3x2x2得矩阵,得到一个2x2的矩阵,其中对应元素是通道间的均值
import numpy as np
a = np.array([
[[1, 2],
[3, 4]],
[[5,6],
[7,8]],
[[1, 2],
[3, 4]]
])
print(a.shape) # (3, 2, 2)
print(np.mean(a, axis=0))
"""
[[2.33333333 3.33333333]
[4.33333333 5.33333333]]
"""
print(np.mean(a, axis=1))
"""
[[2. 3.]
[6. 7.]
[2. 3.]]
"""
print(np.mean(a, axis=2))
"""
[[1.5 3.5]
[5.5 7.5]
[1.5 3.5]]
"""
1、np.mean(x,axis=())

2、先reshape,再求均值(np.mean())

可是求出来的不一样,但是最后的三种均值的sum是相同的。
以后统一用一种方法,出bug在试一下另一种方法是否可行。
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:mmqy2019@163.com进行举报,并提供相关证据,查实之后,将立刻删除涉嫌侵权内容。
长按识别二维码并关注微信
更方便到期提醒、手机管理