了解下 numpy 三维布局

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) 参数的顺序！