У меня есть 5-мерный массив numpy. Я хочу применить преобразование к подмассивам в массиве, как показано ниже
for i in range(N):
for j in range(M):
arr[i, :, j, :, :] = fn(arr[i, :, j, :, :], *args)
Функция (fn) возвращает массив такой же формы, что и входные данные
Есть лучший способ сделать это ? Как я мог бы эффективно векторизовать это?