Autograd Utility#
This module contains utility functions for automatic differentiation, which includes: - Jacobians - Hessians
- Autograd Utility: Batched
bjacrev()- Autograd Utility: Checks
is_batched()is_functorch_tensor()is_gradtracking()- Autograd Utility: Jacobian
jacrev_compat()- Autograd Utility: Gradcheck
dgradcheck()dgradgradcheck()- Autograd Utility: Hessian
hess_fn_rev()hessian()- Autograd Utility: PyTorch AD functions
fjacrev()fvmap()jacrev()vmap()- Autograd Utility: Loop-based Jacobian
jac()