Macro tensor_comp 

macro_rules! tensor_comp {
    ($expr:expr; $var:ident in $start:expr, $end:expr) => { ... };
    ($expr:expr; $var:ident in $start:expr, $end:expr, step $step:expr) => { ... };
    ($expr:expr; $var:ident in $start:expr, $end:expr, if $cond:expr) => { ... };
    ([$expr:expr; $inner_var:ident in $inner_start:expr, $inner_end:expr]; $outer_var:ident in $outer_start:expr, $outer_end:expr) => { ... };
}

Macro for tensor comprehensions