Function stack 

pub fn stack(arrays: &[Array], axis: impl Axis) -> Array

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

use nuvec_core::{array, ops::stack};

assert_eq!(stack(&[array![1, 2], array![3, 4]], -1).shape(), [2, 2]);