Trait rai_core::Module

source · 
pub trait Module {
    // Required methods
    fn forward(&self, input: &Tensor) -> Tensor;
    fn gather_parameters(&self, out: &mut Vec<Tensor>);

    // Provided methods
    fn parameters(&self) -> Vec<Tensor> { ... }
    fn update(&self, _params: &mut HashMap<usize, Tensor>) { ... }
}

Required Methods§

source

fn forward(&self, input: &Tensor) -> Tensor

source

fn gather_parameters(&self, out: &mut Vec<Tensor>)

Provided Methods§

source

fn parameters(&self) -> Vec<Tensor>

source

fn update(&self, _params: &mut HashMap<usize, Tensor>)

Implementors§