1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

use rand::distributions::Distribution;
use rand_distr::StandardNormal;
use zenu_autograd::Variable;
use zenu_matrix::num::Num;

pub mod layers;

pub trait Layer<T: Num> {
    fn init_parameters(&mut self, seed: Option<u64>)
    where
        StandardNormal: Distribution<T>;
    fn parameters(&self) -> Vec<Variable<T>>;
    fn load_parameters(&mut self, parameters: &[Variable<T>]);
    fn clear_gradients(&self) {
        for parameter in self.parameters() {
            parameter.clear_grad();
        }
    }
    fn call(&self, input: Variable<T>) -> Variable<T>;
    fn shape_check(&self, input: &Variable<T>);
}