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

1 问题

如何理解全连接层？

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

2 方法

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

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

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

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

3 结语

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