ybj14

# Refer to `https://j-towns.github.io/2017/06/12/A-new-trick.html` for math details.
import torch
from torch import nn
from torch import autograd

def get_jvp(net, x, v):
  '''
  Generate jacobian vector product. Requires x.requires_grad()
  and v.requires_grad().
  Args: