笔记:李沐《动手学深度学习》 -- 梯度计算

 导数是切线的斜率
 将导数拓展到向量就是梯度 梯度指向值变化最大的方向 与等高线正交

因为定义x是有梯度的张量, y作为x的函数 可以计算关于x的梯度

x = torch.arange(4.0)
x.requires_grad_(True)
x.grad
y = torch.dot(x,x)

 通过反向传播自动计算y关于x每个分量的梯度

y.backward() # 通过反向传播自动计算y关于x每个分量的梯度
x.grad

x.grad == 2*x

# 默认情况下pytorch会累计梯度 需要清除之前的值
x.grad.zero_()

y  = x.sum()
y.backward()
x.grad

 将某些计算移动到计算图之外

x.grad.zero_()
y = x * x
u = y.detach()  # detach之后变为没有梯度的张量

z = u * x 
z.sum().backward()
x.grad == u

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值