神经网络可以使用torch.nn
包构建。它提供了几乎所有与神经网络相关的功能,例如:
- 线性图层
nn.Linear
,nn.Bilinear
- 卷积层
nn.Conv1d
,nn.Conv2d
,nn.Conv3d
,nn.ConvTranspose2d
- 非线性
nn.Sigmoid
,nn.Tanh
,nn.ReLU
,nn.LeakyReLU
- 池化层
nn.MaxPool1d
,nn.AveragePool2d
Recurrent
网络nn.LSTM
,nn.GRU
- 标准化
nn.BatchNorm2d
Dropout
nn.Dropout
,nn.Dropout2d
Embedding
-nn.Embedding
- 损失函数
nn.MSELoss
,nn.C