Function reduce 

pub fn reduce(
    x: &[f32],
    shape: &[usize],
    axes: &[usize],
    op: &str,
    keepdim: bool,
) -> (Vec<f32>, Vec<usize>)