nn.sequential参数
时间: 2023-09-13 12:13:50 浏览: 220
PyTorch快速搭建神经网络及其保存提取方法详解.pdf
nn.Sequential是一个PyTorch模块,它允许用户按顺序将多个模块连接在一起,以构建神经网络。其参数如下:
参数:
- *args (nn.Module) – 一组nn.Modules,按顺序连接在一起形成神经网络。
示例:
下面是一个简单的例子,演示如何使用nn.Sequential构建一个简单的神经网络:
```
import torch.nn as nn
model = nn.Sequential(
nn.Linear(784, 128),
nn.ReLU(),
nn.Linear(128, 10),
nn.LogSoftmax(dim=1)
)
```
在上面的例子中,我们定义了一个包含两个线性层和两个激活函数的神经网络。第一个线性层的输入维度为784,输出维度为128,第二个线性层的输入维度为128,输出维度为10。我们使用ReLU激活函数对第一个线性层的输出进行非线性变换,使用LogSoftmax激活函数对第二个线性层的输出进行标准化处理。
阅读全文