我有一个模型，要求训练图像的形状为 [None, 28, 28, 1]，但 MNIST 图像为 [784,]。如何将训练和测试集中所有图像的形状更改为所需的形状？
提前致谢
最佳答案
您可以使用 numpy 或 tensorflow 本身来 reshape 您的输入。例如，如果您的输入张量的形状为 (784,)，您可以像这样 reshape 它:
import tensorflow as tf
reshaped_tensor = tf.reshape(input_tensor, [28,28,1])
关于python - 改变 tensorflow CNN 的 MNIST 形状，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/50549859/