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
extern crate rand;
pub use self::rand::prelude::*;
#[macro_export]
macro_rules! zeros {
( $n:expr ) => {
vec![0f64; $n]
};
( $r:expr, $c:expr ) => {{
let (r, c) = ($r, $c);
matrix(vec![0f64; r * c], r, c, Row)
}};
}
#[macro_export]
macro_rules! rand {
() => {{
let mut rng = thread_rng();
rng.gen_range(0f64, 1f64)
}};
( $m:expr, $n:expr ) => {{
let r = $m;
let c = $n;
let mut rng = thread_rng();
let mut m = matrix(vec![0f64; r * c], r, c, Row);
for i in 0..r {
for j in 0..c {
m[(i, j)] = rng.gen_range(0f64, 1f64);
}
}
m
}};
}
#[macro_export]
macro_rules! eye {
( $n:expr ) => {{
let n = $n;
let mut m = matrix(vec![0f64; n * n], n, n, Row);
for i in 0..n {
m[(i, i)] = 1f64;
}
m
}};
}
#[macro_export]
macro_rules! linspace {
( $start:expr, $end:expr, $length: expr) => {{
let step = ($end - $start) as f64 / ($length as f64 - 1f64);
seq!($start, $end, step)
}};
}