Function f

intermediate_source/jacobians_hessians.py:332–333

(x)

# Here's some sample usage.

def f(x):
  return x.sin().sum()

x = torch.randn(2048)
tangent = torch.randn(2048)