self.fc1 = nn.Sequential( nn.Linear(in_features=12*7*7, out_features=196), nn.ReLU(), nn.MaxPool2d(kernel_size=2) )补充后续两个全连接层
时间: 2024-03-25 19:39:17 浏览: 51
全连接层FCN的visio图
好的,可以补充两个全连接层如下:
```
self.fc2 = nn.Sequential(
nn.Linear(in_features=196, out_features=64),
nn.ReLU()
)
self.fc3 = nn.Sequential(
nn.Linear(in_features=64, out_features=num_classes),
nn.Softmax(dim=1)
)
```
其中,`nn.Linear`是全连接层模块,参数`in_features`表示输入特征数,`out_features`表示输出特征数;`nn.ReLU`是激活函数模块,使用ReLU函数进行非线性变换;`nn.Softmax`是归一化函数模块,将输出转换为概率分布。在最后一层输出时,一般使用`nn.Softmax`进行归一化处理。
阅读全文