pytorch之Sequential 和 Module

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

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

Sequential模型

Module

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

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


©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容