Struct auto_diff::var::Module

pub struct Module { /* fields omitted */ }

Methods

impl Module

Network holder.

pub fn new() -> Module

Create an empty module. A module is mainly used to create new variables.

pub fn var(&mut self) -> Var

Create a new variable.

pub fn eval(&self)

Try best evaluation of the computation graph.

pub fn forward(&self)

Same as eval

pub fn backward_vector(&self, og: &BTreeMap<NetIndex, Tensor>)

Back propagation

pub fn backward(&self, og: f32)

Back propgation with a single value.

pub fn _visit_data<F>(&self, closure: F) where
    F: Fn(&Op), 

iterator over all data node.

pub fn _visit_op<F>(&self, closure: F) where
    F: Fn(&Op), 

iterator over all op node.