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.