pub fn flatten<S>(a: ArrayBase<S, Ix2>) -> ArrayBase<S, Ix1> where
    S: Data