课程学习目标:构建一个用于Mnist数据集分类的神经网络模型。
关于网络构建
网络构建是指在计算机网络中,通过编程、配置硬件设备或其他方式生成、修改和管理网络的过程。
神经网络模型是由神经网络层和Tensor操作构成的。在MindSpore框架中,神经网络模型的构建是以Cell
类为基础的。一个神经网络模型表示为一个Cell
,它由不同的子Cell
构成。一个复杂的神经网络模型可以被看作是由多个Cell
对象(子Cell)组成的嵌套结构。每个Cell
可以包含其他Cell
作为其子组件,从而允许构建任意复杂的网络架构。使用这样的嵌套结构,可以简单地使用面向对象编程的思维,对神经网络结构进行构建和管理。
实战
定义一个继承自nn.Cell
的类,并在其中构建网络结构。
使用nn.SequentialCell
来替代手动定义多个层并串联它们。
添加Softmax
使用model.parameters_and_names()
来获取模型中的所有参数及其名称。