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 |