美文网首页
pytorch之Sequential 和 Module

pytorch之Sequential 和 Module

作者: 很菜_嗯 | 来源:发表于2019-03-26 15:02 被阅读0次

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

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

    Sequential模型

    Module

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

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


    相关文章

      网友评论

          本文标题:pytorch之Sequential 和 Module

          本文链接:https://www.haomeiwen.com/subject/pjigvqtx.html