Я пытаюсь векторизовать это выражение:
np.vstack([np.dot(arr3d[k], arr2d.T[k]) for k in range(arr3d.shape[0])]).
Это расширение matrix.vector для куба. (Матрица векторов)
Могу ли я заменить k-цикл числовыми индексами, чтобы выполнить ту же операцию?
arr3d - это массив NxNxN, а arr2d - это массив NxN.
Я безуспешно пытаюсь использовать np.einsum и np.indices
Я ожидаю в точности результата np.vstack([np.dot(arr3d[k], arr2d.T[k]) for k in range(arr3d.shape[0])]), то есть матрица NxN = куб NxNxN. (Матрица векторов NxN)