Rust pyo3:有什么方法可以通过引用将参数传递给 pyfunction?

问题描述

fn gen_data<'py> (py: Python<'py>,a: Vec<u8>,b: Vec<u8>) -> PyResult<&'py PyBytes>{
    let dp = MyDataStruct {
        data: Bytes::from(a),extra_info: Bytes::from(b),};
    Ok(PyBytes::new(py,&bincode::serialize(&dp).unwrap()[..]))
}

这是我的一段代码,它当然有效。但就速度而言,这是无法忍受的。我怀疑是复制过程使它变慢。 在我的机器上,我通过 numpy 将 5120x 5120 矩阵转换为字节并传递给它,这需要 3 秒才能完成作业。由于我的服务器非常好,我不希望这种工作需要很长时间

这是我的数据结构定义。

pub struct MyDataStruct {
    pub data: bytes::Bytes,pub extra_info: bytes::Bytes,}

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)