0
点赞
收藏
分享

微信扫一扫

如何进行全连接层代码编写


问题

如何理解全连接层?如何进行全连接层代码编写?

方法

步骤:导入需要用到的包,编写一个类继承nn.Module,将图像用nn.Flatten进行拉伸,编写代码将全连接层连接,输出结果

  1. 导入torch库,和torch里面的nn库,后面要继承nn里面的类
  2. 如何进行全连接层代码编写_深度学习

  3. 将图像进行拉伸,编写全连接层的各层

如何进行全连接层代码编写_python_02

3.定义一个函数forward,先将先进行拉伸,可以输出layer1,layer2,layer3,最后将layer用out表示,输出out

如何进行全连接层代码编写_开发语言_03

结语

针对该问题使用了使用了torch以及torch中的nn.Module类,继承了此类,对图像进行拉伸是此问题的新颖之处,如果图像本就是一维的,则可以省略拉伸这步,对全连接层的理解对全连接层的代码编写极其重要。

举报

相关推荐

0 条评论