pytorch 加入全连接层

在 PyTorch 中加入全连接层可以使用 torch.nn.Linear() 函数。该函数的输入参数为输入特征数和输出特征数，例如：

import torch.nn as nn

# 输入特征数为10，输出特征数为5
fc = nn.Linear(10, 5)

这样就创建了一个输入特征数为10，输出特征数为5的全连接层。