如何从 C++ 端创建一个 numpy 数组并将其提供给 python?
当 Python 不再使用返回的数组时,我希望 Python 进行清理。
C++ 端不会用于delete ret;
释放由new double[size];
.
以下是正确的吗?
#include "pybind11/pybind11.h"
#include "pybind11/numpy.h"
namespace py = pybind11;
py::array_t<double> make_array(const py::ssize_t size) {
double* ret = new double[size];
return py::array(size, ret);
}
PYBIND11_MODULE(my_module, m) {
.def("make_array", &make_array,
py::return_value_policy::take_ownership);
}