rai_opt::optimizers

Trait Optimizer

Source 
pub trait Optimizer {
    // Required method
    fn step(&mut self, grads: &HashMap<usize, Tensor>) -> HashMap<usize, Tensor>;
}

Required Methods§

Source

fn step(&mut self, grads: &HashMap<usize, Tensor>) -> HashMap<usize, Tensor>

return new value for the parameters

Implementors§