airs_types/layers/
mod.rs

1use crate::Tensor;
2
3#[allow(unused_variables)]
4pub trait Layer {
5    fn forward(&self, xs: &Tensor) -> Tensor;
6    fn train_backward(&self, xs: &Tensor, ys: &Tensor) -> Tensor {
7        panic!("This layer does not support backward propagation");
8    }
9    fn train_forward(&self, xs: &Tensor) -> Tensor {
10        self.forward(xs)
11    }
12}