Pytorch 如何所以Variable,Pytorch调用torch.autograd工具包

Song618 次浏览0个评论2018年07月11日
描述

pytorch的自动求导工具包在torch.autograd中[code]from torch.autograd import Variable x = torch.rand(5) x = Variable(x,requires_grad = True) y = x * 2 grads = torch.FloatTensor([1,2,3,4,5]) y.backward(grads)#如果y是scalar的话,那么直接y.backward(),然后通过x.grad方式,就可以得到var的梯度 x.grad #如果y不是scalar,那么只能通过传参的方式给x指定梯度[/code]输出 [code]Variable containing: 2 4 6 8 10 [torch.FloatTensor of size 5][/code]

提交评论

请登录后评论

用户评论

    当前暂无评价,快来发表您的观点吧...

更多相关好文

    当前暂无更多相关好文推荐...

    公告提示

    Pytorch是Facebook的AI研究团队发布了一个Python工具包,是Python优先的深度学习框架。作为numpy的替代品;使用强大的GPU能力,提供最大的灵活性和速度,实现了机器学习框架Torch在Python语言环境的执行,基于python且具备强大GPU加速的张量和动态神经网络。