张量流CNN的MNIST形状变化

Question

我有一个模型，它要求训练图像是形状的[None，28，28，1]，但MNIST图像是[784，]。如何将训练和测试集中的所有图像的形状更改为所需的形状？在

提前谢谢

Answer 1

您可以使用numpy或tensorflow本身来重塑输入。例如，如果您的输入张量是(784,)的形状，则可以按如下方式重塑它：

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