Module autograd::sgd
[−]
[src]
Modules
optimizers |
Functions
apply_gradients |
Update params with gradients |
eval_gradients | |
maybe_reduce_grad |
Reduces gradient's each dim by summation. This is used when parameter shape and gradient shape are not same due to broadcast. |