pybind11和numpy进行交互

使用一个遵循buffer protocol的对象就可以和numpy交互了.

 

这个buffer_protocol要有哪些东西呢? 要有如下接口:

 

struct buffer_info {
    void *ptr;
    ssize_t itemsize;
    std::string format;
    ssize_t ndim;
    std::vector<ssize_t> shape;
    std::vector<ssize_t> strides;
};

 

 

其实就是一个指向数组的指针+各个维度的信息就可以了. 然后我们就可以用指针+偏移来访问数字中的任意位置上的数字了.

 

下面是一个可以跑的例子:

 

 1 #include <pybind11/pybind11.h>
 2 #include <pybind11/numpy.h>
 3 
 4 namespace py = pybind11;