为张量流 CNN 改变 MNIST 的形状
Changing shape of MNIST for tensorflow CNN
我有一个模型要求训练图像的形状为 [None, 28, 28, 1] 但 MNIST 图像是 [784,]。如何将训练和测试集中所有图像的形状更改为所需的形状？
提前致谢
您可以使用 numpy 或 tensorflow 本身来重塑您的输入。例如，如果您的输入张量的形状为 (784,)，您可以像这样重塑它：
import tensorflow as tf
reshaped_tensor = tf.reshape(input_tensor, [28,28,1])
我有一个模型要求训练图像的形状为 [None, 28, 28, 1] 但 MNIST 图像是 [784,]。如何将训练和测试集中所有图像的形状更改为所需的形状？
提前致谢
您可以使用 numpy 或 tensorflow 本身来重塑您的输入。例如，如果您的输入张量的形状为 (784,)，您可以像这样重塑它：
import tensorflow as tf
reshaped_tensor = tf.reshape(input_tensor, [28,28,1])