Допустим, у меня есть 4-мерная матрица A формы [2,4,5,5] в форме numpy.Я хочу поиграть со вторым измерением, т. Е. 4. У меня есть двумерный массив idx=[[1,2],[0,3]].
. Я хочу, чтобы вывод был примерно таким:
B= A[:,idx,:,:] т.е. B должно бытьформы [2,2,5,5].
Есть предложения о том, как этого добиться эффективно?