在过去的几天里,我一直在使用pybind11为现有的 C++ 库创建 Python 绑定,我真的很喜欢它!
可悲的是,我刚刚遇到了一个小问题......
我正在尝试做两件事:
type_caster
将第三方向量类型转换为 NumPy 数组并返回的自定义返回此类型的函数,该类型由以下方式自动矢量化
py::vectorize()
这两件事对自己的工作都很好。带有标量输入的矢量化函数也可以很好地工作。
但是,如果我以数组作为输入调用向量化函数,则会引发异常:
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
RuntimeError: NumPy type info missing for 3vecIdLi2EE
我究竟做错了什么?
或者这根本不应该工作?
以下是我的代码减少到最低限度。在我的实际代码中,vec
该类是第三方库的一部分,并且return_vector()
在我自己的代码中。
mylib.cpp
:
#include <pybind11/pybind11.h>
#include <pybind11/numpy.h>
namespace py = pybind11;
template<typename T, int N> struct vec {
explicit vec(const T* data_) {
for (int i = 0; i < N; ++i) { this->data[i] = data_[i]; }
}
T data[N];
};
vec<double, 2> return_vector(double t) {
double v[] = {t, t};
return vec<double, 2>{v};
}
namespace pybind11 { namespace detail {
template <typename T, int N> struct type_caster<vec<T, N>>
{
private:
using _vecTN = vec<T, N>;
public:
PYBIND11_TYPE_CASTER(_vecTN, _("vec<T, N>"));
bool load(py::handle src, bool convert)
{
if (!convert && !py::array_t<T>::check_(src)) { return false; }
auto buf = py::array_t<T>::ensure(src);
if (!buf || buf.ndim() != 1 || buf.size() != N) { return false; }
value = _vecTN{buf.data()};
return true;
}
static py::handle cast(const _vecTN& src,
py::return_value_policy policy, py::handle parent)
{
py::array_t<T> a({N});
for (auto i = 0; i < N; ++i) { a.mutable_at(i) = src.data[i]; }
return a.release();
}
};
}}
template struct pybind11::detail::type_caster<vec<double, 2>>;
PYBIND11_MODULE(mylib, m) {
m.def("return_vector", py::vectorize(&return_vector));
}
(请随意评论代码,我可能做错了很多事情。我特别不确定我的type_caster
代码。)
为了完整起见,以下是相应的setup.py
:
from setuptools import setup, Extension
class get_pybind_include(object):
def __init__(self, user=False):
self.user = user
def __str__(self):
import pybind11
return pybind11.get_include(self.user)
ext_modules = [
Extension(
'mylib',
['mylib.cpp'],
include_dirs=[
get_pybind_include(),
get_pybind_include(user=True),
],
language='c++',
),
]
setup(
name='mylib',
ext_modules=ext_modules,
install_requires=['pybind11>=2.2'],
)
我已经编译了扩展模块
python3 setup.py develop
运行此 Python 代码可以正常运行:
>>> import mylib
>>> mylib.return_vector(1)
array([1., 1.])
但是,当我使用数组输入调用它时,出现错误:
>>> mylib.return_vector([2, 3])
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
RuntimeError: NumPy type info missing for 3vecIdLi2EE
我本来希望有一个二维数组,例如:
array([[2., 2.],
[3., 3.]])