【Pytorch实用教程】Pytorch中nn.Sequential的用法
nn.Sequential 是 PyTorch 中用于构建神经网络的一种容器类，它可以按顺序封装多个子模块（层），并依次将输入数据传递给这些子模块。这样可以简化模型的定义，使得代码更加简洁和易读。
文章目录
- 基本用法
- 方法一：直接传递子模块
- 方法二：使用 `OrderedDict`
- 动态构建模型
- 优点
- 注意事项
- 自定义模型类
以下是
nn.Sequential 的详细介绍和使用方法：
基本用法
nn.Sequential 可以通过两种方式定义：
直接传递子模块。- 使用
OrderedDict传递子模块。
方法一：直接传递子模块
import torch
import torch.nn as nn# 定义一个简单的前馈神经网络
model = nn.Sequential(nn.Linear