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









