mxnet.autograd¶
Functions
|
Compute the gradients of heads w.r.t previously marked variables. |
|
Retrieve recorded computation history as Symbol. |
|
Compute the gradients of heads w.r.t variables. |
|
Get status on recording/not recording. |
|
Get status on training/predicting. |
|
Mark NDArrays as variables to compute gradient for autograd. |
|
Returns a scope context to be used in ‘with’ statement for codes that do not need gradients to be calculated. |
|
Returns a scope context to be used in ‘with’ statement in which forward pass behavior is set to inference mode, without changing the recording states. |
|
Returns an autograd recording scope context to be used in ‘with’ statement and captures code that needs gradients to be calculated. |
|
Set status to recording/not recording. |
|
Set status to training/predicting. |
|
Returns a scope context to be used in ‘with’ statement in which forward pass behavior is set to training mode, without changing the recording states. |
Classes
|
Customize differentiation in autograd. |