megengine.autodiff#
>>> from megengine.autodiff import GradManager
Defines a block of operations with customizable differentiation. |
|
GradManager computes gradients or more generally, vector-Jacobian product, by reverse mode automatic differentiation (a.k.a. |