Необычное индексирование NumPy (Python)

Question

У меня есть 3D-массив (A) с формой (64, 64, 10), и у меня есть 2D-массив (B) целых чисел с формой (64, 64)

Теперь я хочу использовать целые числа в B в качестве индексов для третьего измерения в A. Поэтому для каждого (x, y) я хочу значение A[x, y, B[x,y]].

Как я могу сделать это с причудливой индексацией в NumPy?

Ура, Табисо