Single function call handling 1d and 2d numpy arrays and returning sum

[bigfatwhale]

Hi,

I'm experimenting with python calling into rust and doing some manipulation before returning the values. To illustrate what I'm trying to achieve let's use the following fragment trying to calculate sum on the input 1d or 2d numpy array,

    #[pyfn(m)]
    #[pyo3(name = "sum")]
    fn sum<'py>(py: Python<'py>,  x: PyReadonlyArrayDyn<'py, f64>, axis: usize) 

    -> Bound<'py, PyArrayLikeDyn<f64>> {
        let dim = x.as_array().dim();
        // dbg!(x);
        match dim.ndim() {
            1 => {
                let x = x.as_array().into_dimensionality::<Ix1>().unwrap();
                let res = _sum1d(&x.view());
                PyArray0::<f64>::new(py, [res], false)
            }
            2 => {
                let x = x.as_array().into_dimensionality::<Ix2>().unwrap();
                let res = _sum2d(&x.view(), axis);
                res.into_dyn().into_pyarray(py)

                
            }
            _ => { 0.0 }
        }
        
    }

However I'm not able to get the code to compile correctly.

Anyone able to advise how to go about this?

Thanks!

[davidhewitt]

not able to get the code to compile correctly.

Can you be more specific on the errors you're getting?