pytoch-2

计算图

计算图是用于描述运算的有向无环图。
计算图有两个主要元素:结点、边
结点表示数据(向量、矩阵、张量)
边表示运算
叶子结点:用户创建的结点
is_leaf:指示张量是否为叶子节点
在这里插入图片描述
requires_grad :是否需要梯度
grad_fn :记录创建该张量时所用的方法
grad :输出该张量的梯度
.backward() :反向求导
当最终输出Tensor为一个标量(一维张量)时,不需要为backward() 指定参数。但是如果它有更多的元素,则需要指定一个gradient参数,它是形状匹配的张量。

动态图/静态图

动态图:运算与搭建同时进行(灵活易于调节)
静态图:先搭建图,后进行运算(高效,不灵活)

autograd

torch.autograd.backward(
						tensors,
						grad_tensors=None,
						retain_graph=None,
						create_graph=False)

tensors: 用于求导的张量
retain_graph :保存计算图
create_graph : 创建导数计算图,用于高阶求导
grad_tensors:多梯度权重

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值