Изменение формы MNIST для тензорного потока CNN

Question

У меня есть модель, которая требует, чтобы тренировочные образы были в форме [Нет, 28, 28, 1], но изображения MNIST [784,].Как изменить форму всех изображений в обучающем и тестовом наборе на требуемую форму?

Заранее спасибо

Answer 1

Вы можете использовать либо numpy, либо tensorflow для изменения формы ввода.Например, если ваш входной тензор имеет форму (784,), вы можете изменить его следующим образом:

import tensorflow as tf
reshaped_tensor = tf.reshape(input_tensor, [28,28,1])