for example:
a = np.zeros((4, 3, 2, 1)) index = (slice(0, 2, 1), 0, ...) # a[index].shape is (2, 2, 1)
The axis 1 (second axis) of a
a
