技术栈
nn.moduledict
郭庆汝
2 年前
人工智能
·
深度学习
·
nn.sequential
·
nn.modulelist
·
nn.moduledict
深度学习基础知识 nn.Sequential ｜ nn.ModuleList ｜ nn.ModuleDict
net = nn.Sequential(nn.Linear(32, 64), nn.ReLU()) →→只需要将定义的层按照顺序写入括号内就可以了