AbstractLayer

nevermind_neu::layers

Trait AbstractLayer 

Source 
pub trait AbstractLayer: WithParams {

Show 15 methods    // Required methods
    fn layer_type(&self) -> &str;
    fn size(&self) -> usize;
    fn cpu_params(&self) -> Option<CpuParams>;
    fn set_cpu_params(&mut self, lp: CpuParams);
    fn set_input_shape(&mut self, sh: &[usize]);
    fn copy_layer(&self) -> Box<dyn AbstractLayer>;
    fn clone_layer(&self) -> Box<dyn AbstractLayer>;

    // Provided methods
    fn forward_input(&mut self, _input_data: Array2D) -> LayerForwardResult { ... }
    fn forward(&mut self, _input: ParamsBlob) -> LayerForwardResult { ... }
    fn backward(
        &mut self,
        _prev_input: ParamsBlob,
        _input: ParamsBlob,
    ) -> LayerBackwardResult { ... }
    fn backward_output(
        &mut self,
        _prev_input: ParamsBlob,
        _expected: Array2D,
    ) -> LayerBackwardResult { ... }
    fn set_batch_size(&mut self, batch_size: usize) { ... }
    fn metrics(&self) -> Option<&Metrics> { ... }
    fn serializable_bufs(&self) -> &[i32] { ... }
    fn trainable_bufs(&self) -> TrainableBufsIds<'_> { ... }

}

Required Methods§

Source

fn layer_type(&self) -> &str

Source

fn size(&self) -> usize

Source

fn cpu_params(&self) -> Option<CpuParams>

Source

fn set_cpu_params(&mut self, lp: CpuParams)

Source

fn set_input_shape(&mut self, sh: &[usize])

Source

fn copy_layer(&self) -> Box<dyn AbstractLayer>

Source

fn clone_layer(&self) -> Box<dyn AbstractLayer>

Provided Methods§

Source

fn forward_input(&mut self, _input_data: Array2D) -> LayerForwardResult

Source

fn forward(&mut self, _input: ParamsBlob) -> LayerForwardResult

Source

fn backward( &mut self, _prev_input: ParamsBlob, _input: ParamsBlob, ) -> LayerBackwardResult

returns out_values and array of weights

Source

fn backward_output( &mut self, _prev_input: ParamsBlob, _expected: Array2D, ) -> LayerBackwardResult

Source

fn set_batch_size(&mut self, batch_size: usize)

Source

fn metrics(&self) -> Option<&Metrics>

Source

fn serializable_bufs(&self) -> &[i32]

Source

fn trainable_bufs(&self) -> TrainableBufsIds<'_>

Implementors§