摘要: 自动求梯度 在深度学习中,我们经常需要对函数求梯度(gradient)。PyTorch提供的 "autograd" 包能够根据输入和前向传播过程自动构建计算图,并执行反向传播。本节将介绍如何使用autograd包来进行自动求梯度的有关操作。 概念 上一节介绍的 是这个包的核心类,如果将其属性 设置为 阅读全文
posted @ 2020-02-22 22:30 hichens 阅读(1259) 评论(0) 推荐(0) 编辑
摘要: 数据操作 在深度学习中,我们通常会频繁地对数据进行操作。作为动手学深度学习的基础,本节将介绍如何对内存中的数据进行操作。 在PyTorch中, 是存储和变换数据的主要工具。如果你之前用过NumPy,你会发现 和NumPy的多维数组非常类似。然而, 提供GPU计算和自动求梯度等更多功能,这些使 更加适 阅读全文
posted @ 2020-02-22 22:17 hichens 阅读(723) 评论(0) 推荐(0) 编辑