Function stack 

pub fn stack(tensors: &[&Tensor], axis: usize) -> Tensor

Stack tensors along a new axis (NumPy stack): each gains a size-1 axis at axis, then they are concatenated there.