Trait corgi::layer::Layer

pub trait Layer {
    fn forward(&self, x: Array) -> Array;
    fn parameters(&mut self) -> Vec<&mut Array>;
}

A layer of a neural network, which implements a forward, and backward pass.

Required methods

fn forward(&self, x: Array) -> Array

Computes the forward pass of the layer.

fn parameters(&mut self) -> Vec<&mut Array>

Retrieves the parameters of the layer.

Loading content...

Implementors

impl Layer for Dense

fn forward(&self, x: Array) -> Array

fn parameters(&mut self) -> Vec<&mut Array>

Loading content...