使用MindSpore定义神经网络模型需要继承mindspore.nn.Cell,神经网络的各层需要预先在__init__方法中定义,然后重载__construct__方法实现神经网络的前向传播过程。因为输入大小处理成32×32的图片,所以需要用Flatten操作将数据压平为一维向量后给全连接层。全连接层输入大小为32×32,输出是预测0~9中的哪个数字,因此输出大小为10。代码2.4定义了一个三层全连接网络模型。
代码2.4 定义三层全连接网络模型