torch.nn.Sequential是什么
时间: 2024-12-04 18:14:31 浏览: 5
对Pytorch中nn.ModuleList 和 nn.Sequential详解
`torch.nn.Sequential` 是PyTorch库中的一个重要模块,它代表了一个线性结构的模型组件集合[^1]。这个类允许开发者按顺序添加一层层的神经网络层,如卷积层、全连接层等,每个层的输入都会直接传递到下一个层。这样,当我们创建一个Sequential实例时,模型的执行过程就是按照构建时的顺序逐层处理输入。
例如,要构造一个简单的卷积神经网络(CNN)链路,可以这样做:
```python
model = torch.nn.Sequential(
torch.nn.Conv2d(3, 16, kernel_size=3),
torch.nn.ReLU(),
torch.nn.MaxPool2d(kernel_size=2),
torch.nn.Flatten(),
torch.nn.Linear(16 * 14 * 14, 128),
torch.nn.ReLU(),
torch.nn.Linear(128, 10)
)
```
在这个例子中,模型先通过一个卷积层,接着经过ReLU激活,然后池化,扁平化,最后连接两个全连接层。
阅读全文