从零开始学习pytorch之计算图与动态图机制

本文介绍了PyTorch中的计算图概念,包括计算图的作用、节点和边的解释,并通过实例展示了叶子节点和梯度的性质。接着探讨了PyTorch的动态图机制,对比静态图,强调了动态图在灵活性和调试方面的优势。
摘要由CSDN通过智能技术生成

计算图(Computational Graph)

前言

在学习了张量的一系列操作后,我们知道深度学习就是对张量进行计算和操作。随着张量种类和计算的增多,会导致各种各样想不到的问题,比如我们多个操作之间该并行还是顺序执行;如何协同各种底层设备以及如何避免各种冗余的操作等等,这些问题都会影响到我们的运算效率,甚至都会引入到一些不必要的bug,而计算图就是为了解决这一个问题而产生的。

  • 计算图是用来描述运算的有向无环图。

  • 计算图有两个主要元素:节点(Node)和边(Edge),节点表示数据,如向量、矩阵;张量表示运算,如加减乘除卷积等

  • 用计算图表示y = (x+w)*(w+1)
    将上式进行拆分a=x+w; b=w+1; y=a*b

叶子节点

用户创建的节点称为叶子节点,如x和w
在这里插入图片描述

is_leaf:指示张量是否为叶子节点

代码查看叶子节点和梯度
import torch

w = torch.tensor([1.], requires_grad=True)
x = torch.tensor([
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值