Numpy: построение трехмерного массива из одномерного массива

Предположим, что задан одномерный массив A . Есть ли простой способ построить трехмерный массив B , такой, что B [i, j, k] = A [k] для всех i, j, k? Вы можете предположить, что форма B задана, и что B.shape [2] = A.shape [0] .