Function split 

pub fn split<T: Clone>(
    array: &Array<T>,
    indices: &[usize],
    axis: usize,
) -> Result<Vec<Array<T>>>

Split an array into multiple subarrays along a specified axis