Python: как изменить форму 3D-изображений (массив np) в 1D, а затем правильно преобразовать их в 3D? ⇐ Python

[Anonymous]

У меня есть изображения RGB (32 x 32 x 3), сохраненные в виде трехмерных массивов, которые я использую в качестве входных данных для своей нейронной сети (с использованием тензорного потока). Чтобы использовать их в качестве входных данных, я преобразую их в массив 1D np (1 x 3072), используя reshape(1,-1). Когда я закончу обучение своей сети, я хочу изменить форму вывода обратно, но использование reshape(32,32,3), похоже, не дает желаемого результата.

Есть это правильный способ сделать это? Как я могу быть уверен, что все данные вернутся в правильное место?

Подробнее здесь: https://stackoverflow.com/questions/440 ... -them-back