Macro arr_rs::array_arange
source · macro_rules! array_arange { ($n:expr, $m:expr) => { ... }; ($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<i32> = array_arange!(0, 4).unwrap();
assert_eq!(array!([0, 1, 2, 3, 4]).unwrap(), arr);
let arr: Array<i32> = array_arange!(0, 7, 2).unwrap();
assert_eq!(array!([0, 2, 4, 6]).unwrap(), arr);