凡科可以做返利網站嗎移動廣告平臺
torch.nn.Sequential
是 PyTorch 中一個模塊容器,用于將一系列層或模塊按順序連接在一起,簡化前向傳播過程。在 Sequential
中,所有的子模塊會按照添加的順序被執(zhí)行,適合那些有明確順序的神經網絡結構,比如卷積神經網絡、全連接網絡等。
主要特點
- 按順序執(zhí)行: 將多個子模塊按順序組合,前向傳播時依次調用。
- 簡潔代碼: 減少顯式定義
forward
方法的需求,對于簡單的網絡結構,使用Sequential
可以大大簡化代碼。 - 嵌套支持:
Sequential
容器可以嵌套,允許將多個Sequential
容器嵌套在一起。
使用方式
- 直接傳入模塊: 可以通過將模塊按順序傳入
Sequential
。 - 有序字典: 可以使用
OrderedDict
來為每個模塊指定名字。
基本用法
1. 直接傳入模塊
import torch
import torch.nn as nnmodel = nn.Sequential(nn.Linear(10, 20),nn.ReLU(),nn.Linear(20, 5)
)input_tensor = torch.randn(1, 10)
output = model(input_tensor)
print(output)
在這個例