a = np.arange(24).reshape(3,2,4)
看上面这个numpy形成的三维数组!
深度:3
高度:2
宽度:4
下面画了一个三维图示,大家用来理解!

先不说高纬度,就看三维,有一些同学这几天问到我,看了上图应该会有一个清晰的理解。
然后下面再理解下 axis 这个参数的功能:
>>> np.sum(a, axis=0)
array([[24, 27, 30, 33],
[36, 39, 42, 45]])
>>> np.sum(a, axis=1)
array([[ 4, 6, 8, 10],
[20, 22, 24, 26],
[36, 38, 40, 42]])
>>> np.sum(a, axis=2)
array([[ 6, 22],
[38, 54],
[70, 86]])
axis = 0 代表了深度方向
axis = 1 代表了高度方向
axis = 2 代表了宽度方向
也对应了 reshape(3, 2, 4) 参数的顺序!
原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/5184/
转载文章受原作者版权保护。转载请注明原作者出处!