当前位置: 首页 > news >正文

每天五分钟玩转深度学习框架PyTorch:将nn的神经网络层连接起来

本文重点

前面我们学习pytorch中已经封装好的神经网络层,有全连接层,激活层,卷积层等等,我们可以直接使用。

如代码所示我们直接使用了两个nn.Linear(),这两个linear之间并没有组合在一起,所以forward的之后,分别调用了,在实际使用中我们常常将几个神经层组合在一起,这样不仅操作方便,而且代码清晰。

这里介绍一下Sequential()和ModuleList(),它们可以将多个神经网络层组合在一起,这两个函数都是特殊的module,包含子module。ModuleList还可以当作list使用,但是不能直接传入输入。

本文主要学习第二步(模型搭建)

Sequential

写法一:为每一个神经网络层添加一个名字

	net1 = nn.Sequential()net1.add_module('conv', nn.Conv2d(3, 3, 3))net1.add_module('batchnorm', nn.BatchNorm2d(3))net1.add_m

http://www.mrgr.cn/news/22334.html

相关文章:

  • PHP-FPM 远程代码执行漏洞(CVE-2019-11043)复现
  • 想做窗套的业主注意了,要提前测量窗扇合页和墙面的距离
  • 【信号】信号的产生
  • LeetCode 热题 100 回顾3
  • 【C++】STL容器详解【下】
  • kubelet组件的启动流程源码分析
  • 【算法专场】模拟(下)
  • 【机器学习】和【人工智能】在物理学领域的应用以及代码案例分析
  • Spring 源码解读:自定义实现Bean定义的注册与解析
  • 位运算技巧总结
  • 【智能排班系统】缓存组件封装
  • lvgl8.3.6 控件垂直布局 label控件在image控件的下方显示
  • 35天学习小结
  • 掌握Hive函数[3]:从基础到高级应用
  • HCIA--实验十一:单区域OSPF路由实验
  • Leetcode第414周赛第二题:3281. 范围内整数的最大得分
  • 线程相关内容
  • Arrays.sort()方法在Java中的使用:理论与实践
  • exec与system的区别(C语言)
  • JS中给元素添加事件监听器的各种方法详解(包含比较和应用场景)