Trait rai_core::Module

source · 
pub trait Module {
    // Required methods
    fn forward(&self, input: &Tensor) -> Tensor;
    fn parameters(&self) -> Vec<Tensor>;
    fn update(&mut self, params: &BTreeMap<usize, Tensor>);
}

Required Methods§

source

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

source

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

source

fn update(&mut self, params: &BTreeMap<usize, Tensor>)

Implementors§