arr_rs/macros/
identity.rs

1/// Create new 2d array with ones on the diagonal and zeros elsewhere
2///
3/// # Examples
4///
5/// ```
6/// use arr_rs::prelude::*;
7///
8/// let arr = array_identity!(i32, 2);
9/// assert_eq!(vec![2, 2], arr.get_shape().unwrap());
10/// let arr = array_identity!(i32, 4);
11/// assert_eq!(vec![4, 4], arr.get_shape().unwrap());
12/// ```
13#[macro_export]
14macro_rules! array_identity {
15    (Tuple2<$t1:ty, $t2:ty>, $($tt:tt)*) => {{
16        compile_error!("only `Numeric` types are supported")
17    }};
18    (Tuple3<$t1:ty, $t2:ty, $t3:ty>, $($tt:tt)*) => {{
19        compile_error!("only `Numeric` types are supported")
20    }};
21    (List<$t1:ty>, $($tt:tt)*) => {{
22        compile_error!("only `Numeric` types are supported")
23    }};
24    (char, $($tt:tt)*) => {{
25        compile_error!("only `Numeric` types are supported")
26    }};
27    (String, $($tt:tt)*) => {{
28        compile_error!("only `Numeric` types are supported")
29    }};
30    ($tt:ty, $n:expr) => {
31        Array::<$tt>::identity($n)
32    };
33}
arr_rs/macros/identity.rs

arr_rs/macros/
identity.rs
