numpy中多维数组的轴（axis）

>>> np.random.seed(123)
>>> X = np.random.randint(0, 5, [3, 2, 2])
>>> print(X)

[[[5 2]
  [4 2]]

 [[1 3]
  [2 3]]

 [[1 1]
  [0 1]]]

>>> X.sum(axis=0)
array([[7, 6],
       [6, 6]])

>>> X.sum(axis=1)
array([[9, 4],
       [3, 6],
       [1, 2]])

>>> X.sum(axis=2)
array([[7, 6],
       [4, 5],
       [2, 1]])

可以将X看成是3×2的矩阵，其中每个元素的长度是2。展成一个二维的平面矩阵。