高阶导数
关注
分享
扫一扫
小然_ran
南京理工与信工所博士，喜欢机器学习，深度学习
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
基于pytorch求自构造函数的高阶导数
import torchfrom torch.autograd import Variabledef f(x): y = x ** 2 return yx = 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 · 1238 阅读 · 0 评论 -
基于pytorch实现牛顿迭代法求函数极小值
import torchfrom torch.autograd import Variabledef f(x): y = x ** 2 return yx = 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 · 1906 阅读 · 0 评论