Layer

Trait Layer 

Source
pub trait Layer {
    // Required method
    fn forward(&self, xs: &Tensor) -> Tensor;

    // Provided methods
    fn train_backward(&self, xs: &Tensor, ys: &Tensor) -> Tensor { ... }
    fn train_forward(&self, xs: &Tensor) -> Tensor { ... }
}

Required Methods§

Source

fn forward(&self, xs: &Tensor) -> Tensor

Provided Methods§

Source

fn train_backward(&self, xs: &Tensor, ys: &Tensor) -> Tensor

Source

fn train_forward(&self, xs: &Tensor) -> Tensor

Implementors§