TensorFlow-MNIST（class类实现）

import tensorflow as tf
from tensorflow.keras.layers import Dense,Flatten
from tensorflow.keras import Model

mnist = tf.keras.datasets.mnist
(x_train,y_train),(x_test,y_test) = mnist.load_data()
x_train,x_test = x_train/255.0,x_test/255.0

class MnistModel(Model):
    def __init__(self):  #定义了call函数中所用到的层
        super(MnistModel,self).__init__()
        self.flatten = Flatten()
        self.d1 = Dense(128,activation = 'relu')
        self.d2 = Dense(10,activation = 'softmax')
        
    def call(self,x): #从输入x到输出y，走过一次前向传播返回输出y
        x = self.flatten(x)
        x = self.d1(x)
        y = self.d2(x)
        return y

model = MnistModel() #实例化model

model.compile(optimizer = 'adam',
              loss = tf.keras.losses.SparseCategoricalCrossentropy(from_logits=False),
              metrics = ['sparse_categorical_accuracy']
              )

model.fit(x_train,y_train,batch_size=32,epochs=5,validation_data=(x_test,y_test) ,validation_freq=1)
model.summary()