PyTorch计算图

最新推荐文章于 2024-01-30 16:06:19 发布

nothing-xianm

最新推荐文章于 2024-01-30 16:06:19 发布

阅读量415

点赞数

分类专栏： pytorch 文章标签：深度学习

原文链接：https://blog.csdn.net/devcy/article/details/89045082

版权

pytorch 专栏收录该内容

3 篇文章 1 订阅

订阅专栏

PyTorch计算图
计算图(Computational Graph)
计算图(Computational Graph)
计算图是计算代数中的一个基础处理方法，我们可以通过一个有向图来表示一个给定的数学表达式，并可以根据图的特点快速方便对表达式中的变量进行求导。而神经网络的本质就是一个多层复合函数, 因此也可以通过一个图来表示其表达式。

#torch.autograd.Variable 实现自动求导
"""
本质上Variable和Tensor没有什么区别，
不过Variable会放在一个计算图里面，
可以进行前向传播和反向传播以及求导
"""
import torch
from torch.autograd import Variable
# requires_grad 表示是否对其求梯度，默认是False
x = Variable(torch.Tensor([3]), requires_grad=True)
y = Variable(torch.Tensor([5]), requires_grad=True)
z = 2 * x * x + y * y + 4
# 对 x 和 y 分别求导
z.backward()
# x 的导数和 y 的导数
#x =3*2*2
#y = 5*2
print('dz/dx: {}'.format(x.grad.data))
print('dz/dy: {}'.format(y.grad.data))

————————————————
版权声明：本文为CSDN博主「devcy」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/devcy/article/details/89045082