网络构建

定义模型类

我们定义神经网络时，可以继承nn.Cell类，在__init__方法中进行子Cell的实例化和状态管理，在construct方法中实现Tensor操作。

常用模型层：

• Flatten: 打平
• Dense: 为全连接层，其使用权重和偏差对输入进行线性变换。
• ReLu：激活函数
• Conv1D： 一维卷积网络
• Conv2D：二维卷积网络
• SequentialCell： 是一个有序的Cell容器。输入Tensor将按照定义的顺序通过所有Cell。
• Softmax： 最后使用nn.Softmax将神经网络最后一个全连接层返回的logits的值缩放为[0, 1]，表示每个类别的预测概率。axis指定的维度数值和为1。