为 TensorFlow 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])