对于Pytorch autograd的一点理解

      刚刚入门深度学习,仅谈论我自己对与autograd的一点认识,如有错误还请大佬指正

      autograd 的数学基础就是链式法则,学过高数的应该多多少少有一定概念,这里也就不在多说了

     Pytorch在autograd的时候主要分为两步走,第一步也就是forward pass 也就是前向传播,第二步是backward pass 也就是反向传播

     在前向传播时主要干了两件事

  1. 一步一步计算网络的结果,并进行保存
  2. 会把当前进行操作的导函数保存到变量的grad_fn这个属性中,之前进行的操作的导函数保存到此变量的grad_fn属性的next_function属性中。

   在进行反向传播的时候主要干三件事

  1. 根据储存的值和导函数算出梯度
  2. 在各自张量的grad属性中累计
  3. 根据链式法则计算出输入对于输出的导

 用一个具体的例子来说明:

对于Z=4(2u+2)^{2}+10(w-3)^{2} 这样一个公式 要求 w对于Z的导就是这样一个流程

代码实现:

def net(x,y):
    x = x*2
    x = x+2
    x = x**2
    x = 4*x
    y = y-3
    y = y**2
    y = y*10
    return x+y

if __name__ =="__main__":
    x = torch.tensor([3.0],requires_grad=True)
    y = torch.tensor([6.0],requires_grad=True)
    out = net(x,y)
    out.backward()
    print(x.grad)

结果:

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值