Function flatten 

pub fn flatten<S>(
    a: ArrayBase<S, Dim<[usize; 2]>>,
) -> ArrayBase<S, Dim<[usize; 1]>>
where
    S: Data,