Macro array_eye

macro_rules! array_eye {
    (Tuple2<$t1:ty, $t2:ty>, $($tt:tt)*) => { ... };
    (Tuple3<$t1:ty, $t2:ty, $t3:ty>, $($tt:tt)*) => { ... };
    (List<$t1:ty>, $($tt:tt)*) => { ... };
    (char, $($tt:tt)*) => { ... };
    (String, $($tt:tt)*) => { ... };
    ($tt:ty, $n:expr) => { ... };
    ($tt:ty, $n:expr, $m:expr) => { ... };
    ($tt:ty, $n:expr, $m:expr, $k:expr) => { ... };
}

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());