Function hvp

intermediate_source/jacobians_hessians.py:326–327

(f, primals, tangents)

from torch.func import jvp, grad, vjp

def hvp(f, primals, tangents):
  return jvp(grad(f), primals, tangents)[1]

#######################################################################
# Here's some sample usage.