在构建比较大的网络层结构时,比如100层网络结构,这个时候再去手动定义参数就显得非常麻烦,所以 PyTorch 提供了两个模块来帮助我们构建模型,一个是Sequential,一个是 Module。
Sequential 允许我们构建序列化的模块,而 Module 是一种更加灵活的模型定义方式。
Sequential模型

Module

将整个网络层的结构操作图表达出来,即前馈神经网络。在__init__中定义每层的结构,在forward中定义前向传播的路径。
下面我们照着模板实现一下上面的神经网络



网友评论