Trait Optimizer 

pub trait Optimizer {
    // Required method
    fn step(
        &mut self,
        params: &[Tensor],
        grads: &[Tensor],
    ) -> Result<Vec<Tensor>>;
}

Optimizer trait: apply one step, returning updated parameters.

Required Methods

fn step(&mut self, params: &[Tensor], grads: &[Tensor]) -> Result<Vec<Tensor>>

Implementors

impl Optimizer for AdamW

impl Optimizer for Sgd