Pytorch的Autograd模块:自动微分

一.写在前面:

        深度学习的算法本质上是通过反向传播求导数,PyTorch的Autograd模块实现了此功能。在Tensor上的所有操作,Autograd都能为他们自动提供微分。

        autograd.Variable是Autograd中的核心类,它简单封装了Tensor,并支持几乎所有Tensor操作。Tensor在被封装为Variable之后,可以调用它的.backwoard实现反向传播,自动计算所有梯度。

          Variable的三个属性。1.data:保存Variable所包含的Tensor。

                                             2.grad:保存data对应的梯度,grad也是个Variable,而不是Tensor,它和data的形状一样。

                                             3.grad_fn:指向一个Function对象,这个Functino用来反向传播计算输入的梯度。

二:

    构建Variable,要注意传入一个参数requires_grad=True,这个参数表示是否对这个变量求梯度,默认的是False,也就是不对这个变量求梯度,这里我们希望得到这些变量的梯度。

      

 上面的代码有一行y.backward(),等价于y.backward(torch.FloatTensor([1])),只不过对于标量求导,里面的参数就不用写了。

 

 

 

posted @ 2019-06-29 22:21  ch_musk  阅读(180)  评论(0编辑  收藏  举报