gpt4 book ai didi

python - 为 tensorflow CNN 改变 MNIST 的形状

转载 作者:太空宇宙 更新时间:2023-11-04 02:25:16 25 4
gpt4 key购买 nike

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

提前致谢

最佳答案

您可以使用 numpytensorflow 本身来 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/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com