基於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)
print(grad_x)

grad_grad_x = torch.autograd.grad(grad_x[0], x)
print(grad_grad_x)