Trait zenu_layer::Layer

pub trait Layer<T: Num> {
    // Required methods
    fn init_parameters(&mut self, seed: Option<u64>)
       where StandardNormal: Distribution<T>;
    fn parameters(&self) -> Vec<Variable<T>>;
    fn load_parameters(&mut self, parameters: &[Variable<T>]);
    fn call(&self, input: Variable<T>) -> Variable<T>;
    fn shape_check(&self, input: &Variable<T>);

    // Provided method
    fn clear_gradients(&self) { ... }
}

Required Methods

fn init_parameters(&mut self, seed: Option<u64>)

where StandardNormal: Distribution<T>,

fn parameters(&self) -> Vec<Variable<T>>

fn load_parameters(&mut self, parameters: &[Variable<T>])

fn call(&self, input: Variable<T>) -> Variable<T>

fn shape_check(&self, input: &Variable<T>)

Provided Methods

fn clear_gradients(&self)

Implementors

impl<T: Num> Layer<T> for Conv2d<T>

impl<T: Num> Layer<T> for Linear<T>