欢迎访问 生活随笔!

生活随笔

当前位置: 首页 > 编程资源 > 编程问答 >内容正文

编程问答

pytorch 学习笔记:nn.Sequential构造神经网络

发布时间:2025/4/5 编程问答 31 豆豆
生活随笔 收集整理的这篇文章主要介绍了 pytorch 学习笔记:nn.Sequential构造神经网络 小编觉得挺不错的,现在分享给大家,帮大家做个参考.

1 两种等价的构造方式

1.1 构建Net类:

1.2 使用torch.nn.Sequential

nn.Sequential里面的模块按照顺序进行排列的,所以必须确保前一个模块的输出大小和下一个模块的输入大小是一致的。

这两种表达方式唯一的不同就是,在1.2中的方法中,激活函数也独立算作一层。

OrderedDict

nn.Sequential中可以使用OrderedDict来指定每个module的名字,而不是采用默认的命名方式(按序号 0,1,2,3...)。

总结

以上是生活随笔为你收集整理的pytorch 学习笔记:nn.Sequential构造神经网络的全部内容,希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错,欢迎将生活随笔推荐给好友。