Macro stack

macro_rules! stack {
    ($axis:expr, $($tensor:expr),+,) => { ... };
    ($axis:expr, $($tensor:expr),+) => { ... };
}