numeric 0.1.4

N-dimensional matrix class for 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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119

macro_rules! add_impl {
    ($t:ty, $m:ident) => (
        mod $m {
            use numeric::{Tensor, AxisIndex};
            type T = Tensor<$t>;

            #[test]
            fn mv_tensor_rf_tensor_2() {
                let t1 = T::new(vec![ 0., 3., 2., 10., -3., 0.]).reshape(&[2, 3]);
                let t2 = T::new(vec![-2., 1., 8.,  1., -1., 2.]).reshape(&[2, 3]);
                let answer = T::new(vec![2., 2., -6., 9., -2., -2.]).reshape(&[2, 3]);
                assert!(t1.clone() - &t2 == answer);
            }

            #[test]
            fn rf_tensor_rf_tensor_3() {
                let t1 = T::new(vec![ 0., 3., 2., 10., -3., 0.]).reshape(&[2, 3]);
                let t2 = T::new(vec![-2., 1., 8.,  1., -1., 2.]).reshape(&[2, 3]);
                let answer = T::new(vec![2., 2., -6., 9., -2., -2.]).reshape(&[2, 3]);
                assert!(&t1 - &t2 == answer);
            }

            #[test]
            #[should_panic(expected = "assertion failed")]
            fn mv_tensor_rf_tensor_mismatched_1() {
                let t1 = T::new(vec![ 0., 3., 2., 10., -3., 0.]).reshape(&[2, 3]);
                let t2 = T::new(vec![-2., 1., 8.,  0.]).reshape(&[2, 2]);
                t1 - &t2; // mis-matched shape
            }

            #[test]
            fn rf_tensor_rf_tensor_slices_reverse_1() {
                let t1 = T::new(vec![ 0., 3., 2., 10., -3., 0.]).reshape(&[2, 3]);
                let t2 = T::new(vec![-2., 1., 8.,  1., -1., 2.]).reshape(&[2, 3])
                    .index(&[AxisIndex::StridedSlice(None, None, -1)]);
                let answer = T::new(vec![-1., 4., 0., 12., -4., -8.]).reshape(&[2, 3]);
                assert!(&t1 - &t2 == answer);
            }

            #[test]
            fn rf_tensor_rf_tensor_slices_reverse_2() {
                let t1 = T::new(vec![ 0., 3., 2., 10., -3., 0.]).reshape(&[2, 3]);
                let t2 = T::new(vec![-2., 1., 8.,  1., -1., 2.]).reshape(&[2, 3])
                    .index(&[AxisIndex::Full, AxisIndex::StridedSlice(None, None, -1)]);
                let answer = T::new(vec![-8., 2., 4., 8., -2., -1.]).reshape(&[2, 3]);
                assert!(&t1 - &t2 == answer);
            }

            #[test]
            fn mv_tensor_rf_tensor_slices_reverse_1() {
                let t1 = T::new(vec![ 0., 3., 2., 10., -3., 0.]).reshape(&[2, 3]);
                let t2 = T::new(vec![-2., 1., 8.,  1., -1., 2.]).reshape(&[2, 3])
                    .index(&[AxisIndex::StridedSlice(None, None, -1)]);
                let answer = T::new(vec![-1., 4., 0., 12., -4., -8.]).reshape(&[2, 3]);
                assert!(t1 - &t2 == answer);
            }

            #[test]
            fn mv_tensor_rf_tensor_slices_reverse_2() {
                let t1 = T::new(vec![ 0., 3., 2., 10., -3., 0.]).reshape(&[2, 3]);
                let t2 = T::new(vec![-2., 1., 8.,  1., -1., 2.]).reshape(&[2, 3])
                    .index(&[AxisIndex::Full, AxisIndex::StridedSlice(None, None, -1)]);
                let answer = T::new(vec![-8., 2., 4., 8., -2., -1.]).reshape(&[2, 3]);
                assert!(t1 - &t2 == answer);
            }

            #[test]
            fn rf_tensor_rf_tensor_slices_subset_1() {
                let t1 = T::range(20)
                    .index(&[AxisIndex::StridedSlice(Some(2), Some(7), 2)]);
                let t2 = T::range(10)
                    .index(&[AxisIndex::StridedSlice(Some(1), Some(4), 1)]);
                let answer = T::new(vec![1., 2., 3.]);
                assert!(&t1 - &t2 == answer);
            }

            #[test]
            fn rf_tensor_rf_tensor_broadcast_1() {
                let t1 = T::new(vec![10., 20., 40., 5., 10., 20.]).reshape(&[3, 2]);
                let t2 = T::new(vec![2., 4.]);
                let answer = T::new(vec![8., 16., 38., 1., 8., 16.]).reshape(&[3, 2]);
                assert!(&t1 - &t2 == answer);
            }

            #[test]
            fn rf_tensor_rf_tensor_broadcast_2() {
                let t1 = T::new(vec![10., 20., 40., 5., 10., 20.]).reshape(&[3, 2]);
                let t2 = T::new(vec![2., 4., 1.]).index(&[AxisIndex::Full, AxisIndex::NewAxis]);
                let answer = T::new(vec![8., 18., 36., 1., 9., 19.]).reshape(&[3, 2]);
                assert!(&t1 - &t2 == answer);
            }

            #[test]
            fn rf_tensor_rf_tscalar_1() {
                let t1 = T::new(vec![ 0., 3., 2., 10., -3., 0.]).reshape(&[2, 3]);
                let t2 = T::scalar(-2.);
                let answer = T::new(vec![0., -1.5, -1., -5., 1.5, 0.]).reshape(&[2, 3]);
                assert!(&t1 / &t2 == answer);
            }

            #[test]
            fn rf_tensor_scalar_1() {
                let t1 = T::new(vec![ 0., 3., 2., 10., -3., 0.]).reshape(&[2, 3]);
                let answer = T::new(vec![-2., 1., 0., 8., -5., -2.]).reshape(&[2, 3]);
                assert!(&t1 - 2. == answer);
            }

            #[test]
            fn mv_tensor_scalar_1() {
                let t1 = T::new(vec![ 0., 3., 2., 10., -3., 0.]).reshape(&[2, 3]);
                let answer = T::new(vec![-2., 1., 0., 8., -5., -2.]).reshape(&[2, 3]);
                assert!(t1 - 2. == answer);
            }
        }
    )
}

add_impl!(f32, float32);
add_impl!(f64, float64);