arr-rs 0.6.0

arr-rs - rust arrays library
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

/// Create new 2d array with ones on the diagonal and zeros elsewhere
///
/// # Examples
///
/// ```
/// use arr_rs::prelude::*;
///
/// let arr = array_eye!(i32, 2, 3, 3);
/// assert_eq!(vec![2, 3], arr.get_shape().unwrap());
/// let arr = array_eye!(i32, 4, 5, 3);
/// assert_eq!(vec![4, 5], arr.get_shape().unwrap());
/// ```
#[macro_export]
macro_rules! array_eye {
    (Tuple2<$t1:ty, $t2:ty>, $($tt:tt)*) => {{
        compile_error!("only `Numeric` types are supported")
    }};
    (Tuple3<$t1:ty, $t2:ty, $t3:ty>, $($tt:tt)*) => {{
        compile_error!("only `Numeric` types are supported")
    }};
    (List<$t1:ty>, $($tt:tt)*) => {{
        compile_error!("only `Numeric` types are supported")
    }};
    (char, $($tt:tt)*) => {{
        compile_error!("only `Numeric` types are supported")
    }};
    (String, $($tt:tt)*) => {{
        compile_error!("only `Numeric` types are supported")
    }};
    ($tt:ty, $n:expr) => {
        array_eye!($tt, $n, $n, 0)
    };
    ($tt:ty, $n:expr, $m:expr) => {
        array_eye!($tt, $n, $m, 0)
    };
    ($tt:ty, $n:expr, $m:expr, $k:expr) => {{
        Array::<$tt>::eye($n, Some($m), Some($k))
    }};
}