不带拷贝的二维向量pybind11数组

2024-04-23 18:43:05 发布

您现在位置:Python中文网/ 问答频道 /正文

如何从pybind11中的向量向量创建2d numpy数组

我正在寻找一种快速完成此任务的方法&;没有复制(所以移动?),我想用模板化的方式来做

到目前为止,我已经:

template< typename T >
py::array
array2d_from_vector(std::vector<std::vector<T>> & m) {
    if (m.empty()) return py::array_t<T>();
    std::vector<std::vector<T>>* ptr = new std::vector<std::vector<T>>(std::move(m));
    auto capsule = py::capsule(ptr, [](void* p) {
            delete reinterpret_cast<std::vector<std::vector<T>>*>(p);
    });
    return py::array_t<T>(
            {ptr->size(), ptr->at(0).size()},           // shape of array
            {ptr->at(0).size()*sizeof(T), sizeof(T)},   // c-style contiguous strides
            capsule);
}

它向python返回一个无意义的2d数组,其值如下:

[[ 4.0852747e+30  3.0677226e-41  4.0853762e+30  3.0677226e-41]...

那么我错过了什么!我猜这与向量保存数据的方式有关,但我的值应该在0到1之间


Tags: pynumpysizereturn方式数组array向量