1,加载数据,加载数据集
2,定义网络,用class nect(nn.module)来调用类的形式来使用神经网络
def __init__(self):
super()__init__()
flatten,将输入的二维图像数据展平为一维向量
hidden,线性层,输入为28*28,输出为128
等等
forward,前向传播函数,定义了模型怎么计算和输出
x=self。flatten
x= self。hidden
x = self。relu
激活函数
用在线性层和隐藏层
return x