Pytorch中的自动求梯度机制和Variable类实例

qq_22382 13 0 PDF 2020-09-29 16:09:54

自动求导机制是每一个深度学习框架中重要的性质,免去了手动计算导数,下面用代码介绍并举例说明Pytorch的自动求导机制。torch.Tensor能像Variable那样追踪历史和反向传播。Variable仍能正确工作,但是返回的是Tensor。我们拥抱这些新特性,看看Pytorch怎么进行自动求梯度。这里添加一个小知识点,即torch.Tensor和torch.tensor的不同。二者均可以生成新的张量,但torch.Tensor()是python类,是默认张量类型torch.FloatTensor()的别名,使用torch.Tensor()会调用构造函数,生成单精度浮点类型的张量。而torch.tensor()是函数,其中data可以是list,tuple,numpy,ndarray,scalar和其他类型,但只有浮点类型的张量能够自动求梯度。torch.tensor言归正传,上一个例子的变量本质上是标量。

Pytorch中的自动求梯度机制和Variable类实例

Pytorch中的自动求梯度机制和Variable类实例

用户评论
请输入评论内容
评分:
暂无评论