InitParameters

rstorch::module::init

Trait InitParameters 

Source 
pub trait InitParameters {
    // Required method
    fn weight(&self, input_size: usize, output_size: usize) -> Array2<f64>;

    // Provided method
    fn bias(&self, _input_size: usize, output_size: usize) -> Array2<f64> { ... }
}

Required Methods§

Source

fn weight(&self, input_size: usize, output_size: usize) -> Array2<f64>

Provided Methods§

Source

fn bias(&self, _input_size: usize, output_size: usize) -> Array2<f64>

Implementors§