多维度的Numpy花式索引

Let's say I have an numpy array A of size n x m x k and another array B of size n x m that has indices from 1 to k. I want to access each n x m slice of A using the index given at this place in B, giving me an array of size n x m.Let's say I have an numpy array A of size n x m