autograd 1.1.1

Tensors and differentiable operations in Rust
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
54

extern crate autograd as ag;
extern crate ndarray;

#[test]
fn random_uniform() {
    let rng = ag::ndarray_ext::ArrayRng::<f64>::default();
    let a = rng.random_uniform(&[3], 0., 1.);
    let b = rng.random_uniform(&[3], 0., 1.);
    assert!(a.all_close(&ndarray::arr1(&[0.5, 0.5, 0.5]), 0.5));
    assert_ne!(a, b);
}

#[test]
fn random_normal() {
    let rng = ag::ndarray_ext::ArrayRng::<f64>::default();
    let a = rng.random_normal(&[3], 0., 1.);
    let b = rng.random_normal(&[3], 0., 1.);
    assert_ne!(a, b);
    assert_eq!(a.shape(), &[3])
}

#[test]
fn standard_normal() {
    let rng = ag::ndarray_ext::ArrayRng::<f64>::default();
    let a = rng.standard_normal(&[3]);
    let b = rng.standard_normal(&[3]);
    assert_ne!(a, b);
    assert_eq!(a.shape(), &[3])
}

#[test]
fn standard_uniform() {
    let rng = ag::ndarray_ext::ArrayRng::<f64>::default();
    let a = rng.standard_uniform(&[3]);
    let b = rng.standard_uniform(&[3]);
    assert!(a.all_close(&ndarray::arr1(&[0.5, 0.5, 0.5]), 0.5));
    assert_ne!(a, b);
}

#[test]
fn glorot_normal() {
    let rng = ag::ndarray_ext::ArrayRng::<f64>::default();
    let a = rng.glorot_normal(&[3, 2]);
    let b = rng.glorot_normal(&[3, 2]);
    assert_ne!(a, b);
}

#[test]
fn glorot_uniform() {
    let rng = ag::ndarray_ext::ArrayRng::<f64>::default();
    let a = rng.glorot_uniform(&[3, 2]);
    let b = rng.glorot_uniform(&[3, 2]);
    assert_ne!(a, b);
}