首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >pytorch中的向后、梯度函数

问pytorch中的向后、梯度函数
EN

Stack Overflow用户
提问于 2017-05-30 21:44:36
回答 0查看 2.4K关注 0票数 4

我正在尝试在pytorch中实现向后的grad函数。

但是,我不知道为什么会返回这个值。

这是我的代码。

代码语言:javascript
复制
x = Variable(torch.FloatTensor([[1,2],[3,4]]), requires_grad=True)
y = x + 2
z = y * y

gradient = torch.ones(2, 2)
z.backward(gradient)
print(x.grad)

我认为结果值应该是[6,8,10,12]

由于dz/dx= 2*(x+2)和x=1,2,3,4

但返回值为[7,9,11,13]

为什么会发生这种情况..我想知道梯度,梯度函数做得怎么样。

请帮帮我。

EN

回答

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/44264018

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档