Trait rai_core::Module

pub trait Module {
    // Required method
    fn forward(&self, input: &Tensor) -> Tensor;

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

Required Methods

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

Provided Methods

fn gather_parameters(&self, params: &mut HashMap<usize, Tensor>)

fn parameters(&self) -> HashMap<usize, Tensor>

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

Implementations on Foreign Types

impl<'a, T> Module for &'a T

where T: Module,

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

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

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

Implementors