1#[macro_export]
2macro_rules! debug_tensor {
3 ($w:ident, $t:ident; $d:literal $(,$ds:literal)+; $($is:ident),*) => {
4 for i in 0..$d {
5 debug_tensor!($w, $t; $($ds),*; $($is,)* i);
6 writeln!($w)?;
7 }
8 };
9 ($w:ident, $t:ident; $d:literal; $($is:ident),+) => {
10 for i in 0..$d {
11 write!($w, "{:?}\t", $t[($($is,)* i)])?;
12 }
13 };
14 ($w:ident, $t:ident; $d:literal; ($i:ident)) => {
15 for i in 0..$d {
16 write!($w, "{:?}\t", $t[($i, i)])?;
17 }
18 };
19 ($w:ident, $t:ident; $d:literal;) => {
20 for i in 0..$d {
21 write!($w, "{:?}\t", $t[i])?;
22 }
23 };
24}