Module autograd::ops::gradient_descent_ops [] [src]

Re-exports

pub use self::adam::Adam;
pub use self::sgd::SGD;

Modules

adam
sgd

Structs

StateKey

Key to access a state tensor. Stateful optimizers use this.

Traits

Optimizer

Trait for any gradient descent optimizer