rotta_rs 0.0.6

a Deep Learning library with rust language
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13

use rand_distr::{ Distribution, Normal };

use crate::rotta_rs_module::Module;

impl Module {
    pub fn he_initialization(&mut self, input: usize) -> f64 {
        let he = 2.0 / (input as f64);
        let he_std = he.sqrt();

        let normal = Normal::new(0.0, he_std).unwrap();
        normal.sample(&mut self.rng)
    }
}