可以用import torch.nn 来使用Sequential()吗
时间: 2024-05-21 09:10:20 浏览: 68
可以。在使用Sequential()时,需要从torch.nn中导入Sequential。例如:
import torch.nn as nn
model = nn.Sequential(
nn.Linear(10, 20),
nn.ReLU(),
nn.Linear(20, 30),
nn.ReLU(),
nn.Linear(30, 2)
)
阅读全文