I have a 3d tensor and I want to select different slices from the dim=2. something like `a[[0, 1], :, [slice(2, 4), slice(1, 3)]]`.

```py
a=np.arange(2*3*5).reshape(2, 3, 5)
array([[[ 0,  1,  2,  3,  4],
        [ 5,  6,  7,  8,  9],
        [10, 11, 12, 13, 14]],

       [[15, 16, 17, 18, 19],
        [20, 21, 22, 23, 24],
        [25, 26, 27, 28, 29]]])
# then I want something like a[[0, 1], :, [slice(2, 4), slice(1, 3)]]
# that gives me np.stack([a[0, :, 2:4], a[1, :, 1:3]]) without a for loop
array([[[ 2,  3],
        [ 7,  8],
        [12, 13]],

       [[16, 17],
        [21, 22],
        [26, 27]]])

```

and I've seen [this](https://stackoverflow.com/questions/43413582/selecting-multiple-slices-from-a-numpy-array-at-once) and it is not what I want.