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)