高阶导数
小然_ran
南京理工与信工所博士,喜欢机器学习,深度学习
展开
-
基于pytorch求自构造函数的高阶导数
import torch from torch.autograd import Variable def f(x): y = x ** 2 return y x = Variable(torch.Tensor([5]), requires_grad=True) grad_x = torch.autograd.grad(f(x), x, create_graph=True) p...原创 2019-05-20 17:00:14 · 1169 阅读 · 0 评论 -
基于pytorch实现牛顿迭代法求函数极小值
import torch from torch.autograd import Variable def f(x): y = x ** 2 return y x = Variable(torch.Tensor([5]), requires_grad=True) for i in range(100): grad_x = torch.autograd.grad(f(x)...原创 2019-05-20 17:23:14 · 1784 阅读 · 0 评论