pytorch之Sequential 和 Module

2019-03-26  本文已影响0人  很菜_嗯

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

Sequential 允许我们构建序列化的模块,而 Module 是一种更加灵活的模型定义方式。

Sequential模型

Module

将整个网络层的结构操作图表达出来,即前馈神经网络。在__init__中定义每层的结构,在forward中定义前向传播的路径。

下面我们照着模板实现一下上面的神经网络


上一篇下一篇

猜你喜欢

热点阅读