индекс 2 находится за пределами оси 0 с размером 2

Question

У меня есть 3-мерный массив numpy, и я хотел бы выполнить перестановку на второй оси. Как я могу это сделать?

def randPerm(A):
    L, N, _ = A.shape
    pA = np.empty_like(A)

    for i in range(L):
        peridx = np.random.permutation(N)
        pA[i, :, :] = A[i, peridx, :]

    return pA

Я выполнил указанную выше функцию на массиве (5, 57, 2) np.array и получил сообщение об ошибке: «индекс 2 находится за пределами оси 0 с размер 2 ".