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