meuron 0.4.0

Meuron is a modular neural network library written in rust for training simple neural networks.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

use crate::backend::Backend;
use crate::initializer::Initializer;
use ndarray::Dimension;

#[derive(Clone, Copy, Debug, Default)]
pub struct XavierUniform;

impl<B: Backend> Initializer<B> for XavierUniform {
    fn init<D: Dimension>(&self, shape: D) -> B::Tensor<D> {
        let dims = shape.slice();
        assert!(
            dims.len() == 2,
            "XavierUniform expects a 2D shape (fan_in, fan_out)"
        );

        let fan_in = dims[0] as f32;
        let fan_out = dims[1] as f32;
        let limit = (6.0 / (fan_in + fan_out)).sqrt();

        B::random_uniform(shape, -limit, limit)
    }
}