Python '；图像数据的无效形状（28,28,1）'；

这是我的代码示例：

    (train_images, train_labels), (test_images, test_labels) = datasets.mnist.load_data()
    
    train_images = train_images.reshape((60000, 28, 28, 1))
    test_images = test_images.reshape((10000, 28, 28, 1))
    
    #train_images, test_images = train_images / 255.0, test_images / 255.0
    train_images = train_images.astype('float32') / 255
    
    test_images = test_images.astype('float32') / 255
    
    plt.imshow(train_images[0])

但是我在/255之前得到了doplt.imshow（火车图像[0][0]）

或者使用train\u images=train\u images[0]/255并保持代码不变

如果您打印火车图像[0]，您可以计算出来。形状显示为60000 x 28 x 28

编辑：添加代码

import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import datasets, layers, models
(train_images, train_labels) = datasets.mnist.load_data()
import matplotlib.pyplot as plt
train_images = train_images[0] / 255
plt.imshow(train_images[0])

---

欢迎来到SO。请不要发布数据/代码/错误消息的图像。将文本张贴在此处。你也应该阅读并吸引好的答案。我建议你编辑你的问题。这能回答你的问题吗？