文档摘要

4.2 nn.Module 构建方式 本节摘要：PyTorch 模型几乎都继承 。 里创建 、 等子模块并赋给 ， 里描述张量怎么流过它们。原文入门分类器： 成 784，Linear 到 128，ReLU，Linear 到 10，输出 logits。 把参数迁到 GPU； 交给 Adam； 与 切换 Dropout 等行为。漏把层挂到 上，优化器就看不见那些权重。 会员。《4.2 nn.Module构建方式》收录于灏天文库文集《TensorFlow、PyTorch 入门实战：快速搭建你的第一个AI模型》，原作者/来源：灏天文库，整理自「灏天文库」，提供技术教程、实践指南与问题解决方案，支持在线阅读、全文检索与知识沉淀，助力开发者系统化学习。

该文档为会员专享，请先登录或注册后再查看

登录 注册

---

作者与出处

原作者： 灏天文库

来源：灏天文库

原文： https://aiknowledge.cn/collection/465-tensorflowpytorch-%E5%85%A5%E9%97%A8%E5%AE%9E%E6%88%98%E5%BF%AB%E9%80%9F%E6%90%AD%E5%BB%BA%E4%BD%A0%E7%9A%84%E7%AC%AC%E4%B8%80%E4%B8%AAai%E6%A8%A1%E5%9E%8B

整理： 灏天文库整理

由灏天文库平台收录，内容或由平台用户上传，仅供学习交流