Установка размеров массива поездов для распознавания лиц для Python

Я пытаюсь найти эквивалентный код для Python, чтобы установить следующие размеры, которые я установил в R:

dim(train_array) = c(28, 28, 1, ncol(train_x))

train_x - это матрица из 28 на 28 изображений лица оливетти с измененным размером (шкала серого), оцифрованных в 784 (28 * 28) строк пикселей (каждая строка помечена как пиксель 1, пиксель 2 и т. д.) и 15 столбцов. Есть ли какой-либо пакет python, который выполняет ту же функцию, что и dim () для R в python?