jactorch.utils.grad#

Functions

no_grad_func(func)

A decorator to disable gradient calculation for a function.

Functions

no_grad_func(func)[source]#

A decorator to disable gradient calculation for a function.