下载掌阅APP，畅读海量书库
立即打开
第2章完成了第一个PyTorch深度学习示例程序——一个非常简单的MNIST手写体生成器，其作用是向读者演示一个PyTorch深度学习程序的基本构建与完整的训练过程。
PyTorch作为一个成熟的深度学习框架，对于使用者来说，即使是初学者，也能很容易地用其进行深度学习项目的训练，只要编写出简单的代码就可以构建相应的模型进行实验，但其缺点在于框架的背后内容都被隐藏起来了。
本章将使用Python实现一个轻量级的、易于扩展的深度学习框架，目的是希望读者从这一过程中了解深度学习的基本组件以及框架的设计和实现，从而为后续的学习打下基础。
本章首先使用PyTorch完成MNIST分类的练习，主要是为了熟悉PyTorch的基本使用流程；之后将实现一个自定义的深度学习框架，从基本的流程开始分析，对神经网络中的关键组件进行抽象，确定基本框架，然后对框架中的各个组件进行代码实现；最后基于自定义框架实现MNIST分类，并与PyTorch实现的MNIST分类进行简单的对比验证。