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

Posted 2022-12-14 算法与编程之美

问题

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

方法

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

1. 导入torch库，和torch里面的nn库，后面要继承nn里面的类
   

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

---

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

结语

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