Function stack 

pub fn stack(arrays: &[Array], axis: isize) -> Result<Array>

arrays (one shape) stacked along a new axis axis (MLX’s stack).