1.先训练数据
import tensorflow as tf
from tensorflow.keras import Sequential, layers, optimizers
# 加载数据集
mnist = tf.keras.datasets.mnist
(trainImage, trainLabel),(testImage, testLabel) = mnist.load_data()
for i in [trainImage,trainLabel,testImage,testLabel]:
print(i.shape)
trainImage = tf.reshape(trainImage,(60000,28,28,1))
testImage = tf.reshape(testImage,(10000,28,28,1))
for i in [trainImage,trainLabel,testImage,testLabel]:<