flashlight/layers/
mod.rs

1use flashlight_tensor::tensor::Tensor;
2
3pub mod linear;
4pub mod dropout;
5
6pub mod activations;
7
8/// Basic layer trait for layers
9pub trait Layer {
10    /// Forward propagation for layer
11    fn forward(&mut self, input: &Tensor<f32>) -> Tensor<f32>;
12    /// Backward propagation for layer
13    fn backward(&mut self, grad_output: &Tensor<f32>) -> Tensor<f32>;
14}