Macro arr_rs::array_arange

source ·
macro_rules! array_arange {
    ($tt:ty, $n:expr, $m:expr) => { ... };
    ($tt:ty, $n:expr, $m:expr, $k:expr) => { ... };
}
Expand description

Create new 2d array with ones on the diagonal and zeros elsewhere

Examples

use arr_rs::prelude::*;

let arr = array_arange!(i32, 0, 4).unwrap();
assert_eq!(array!(i32, [0, 1, 2, 3, 4]).unwrap(), arr);
let arr = array_arange!(i32, 0, 7, 2).unwrap();
assert_eq!(array!(i32, [0, 2, 4, 6]).unwrap(), arr);