神经网络可以使用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.CrossEntropyLoss
,nn.NLLLoss
这些类的实例将具有一个内置的__call__
函数,可通过图层运行输入。
任务描述
本关任务:
本关要求利用nn.Linear()