pytorch图

LGDDDDDD

于 2020-01-22 08:49:13 发布

阅读量157

点赞数

CC 4.0 BY-SA版权

分类专栏： pytorch

本文链接：https://blog.youkuaiyun.com/weixin_43289424/article/details/104067489

pytorch 专栏收录该内容

13 篇文章

订阅专栏

本文通过一个具体的例子展示了如何使用PyTorch进行动态图计算，包括创建张量、定义计算图、执行反向传播以及打印梯度等关键步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

图

计算图
动态图

计算图

在这里插入图片描述

import  torch

w = torch.tensor([1.],requires_grad=True)
x = torch.tensor([2.],requires_grad=True)

a = torch.add(w,x)
a.retain_grad()
b = torch.add(w,1)
y = torch.mul(a,b)

y.backward()
print(w.grad)
print(a.grad)

动态图

在这里插入图片描述