concision-params 0.3.1

this crate implements the core modules for the concision framework
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

/*
    Appellation: params <test>
    Contrib: @FL03
*/
use concision_params::Params;

#[test]
fn test_params_ones() {
    // weights retain the given shape (in, out)
    // bias retains the shape (out,)
    let ones = Params::<f64>::ones((3, 4));
    assert_eq!(ones.dim(), (3, 4));
    assert_eq!(ones.bias().dim(), 4);
    assert!(
        ones.iter()
            .all(|(w, b)| w.iter().all(|&wi| wi == 1.0) && b == &1.0)
    );
}

#[test]
fn test_params_zeros() {
    // weights retain the given shape (in, out)
    // bias retains the shape (out,)
    let zeros = Params::<f64>::zeros((3, 4));
    assert_eq!(zeros.dim(), (3, 4));
    assert_eq!(zeros.bias().dim(), 4);
    assert!(
        zeros
            .iter()
            .all(|(w, b)| w.iter().all(|&wi| wi == 0.0) && b == &0.0)
    );
}

#[test]
#[cfg(feature = "rand")]
fn test_params_init_rand() -> anyhow::Result<()> {
    use concision_init::NdRandom;

    let lecun = Params::<f64>::lecun_normal((3, 4));
    assert_eq!(lecun.dim(), (3, 4));

    let glorot_norm = Params::<f64>::glorot_normal((3, 4));
    assert_eq!(glorot_norm.dim(), (3, 4));
    assert_ne!(lecun, glorot_norm);
    let glorot_uniform = Params::<f64>::glorot_uniform((3, 4)).expect("glorot_uniform failed");
    assert_eq!(glorot_uniform.dim(), (3, 4));
    assert_ne!(lecun, glorot_uniform);
    assert_ne!(glorot_norm, glorot_uniform);
    let truncnorm = Params::<f64>::truncnorm((3, 4), 0.0, 1.0).expect("truncnorm failed");
    assert_eq!(truncnorm.dim(), (3, 4));

    Ok(())
}