Skip to main content

Optimizer

Trait Optimizer 

Source
pub trait Optimizer<B: Backend> {
    // Required method
    fn update_param<D: Dimension>(
        &mut self,
        param: &mut B::Tensor<D>,
        grad: &B::Tensor<D>,
    );
}

Required Methods§

Source

fn update_param<D: Dimension>( &mut self, param: &mut B::Tensor<D>, grad: &B::Tensor<D>, )

Dyn Compatibility§

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementations on Foreign Types§

Source§

impl<B: Backend, O: Optimizer<B>> Optimizer<B> for &mut O

Source§

fn update_param<D: Dimension>( &mut self, param: &mut B::Tensor<D>, grad: &B::Tensor<D>, )

Implementors§

Source§

impl<B: Backend> Optimizer<B> for SGD