C++与pybind11：能否为std::span实现自定义type_caster？

完全可以通过自定义type_caster实现Python numpy数组到C++ std::span<int>的零拷贝传参，核心是利用numpy连续内存的特性直接映射到std::span，无需复制数据。

完整实现代码 #

首先包含必要头文件：

#include <pybind11/pybind11.h>
#include <pybind11/numpy.h>
#include <span>

namespace py = pybind11;

然后实现std::span<int>的type_caster：

namespace pybind11 {
namespace detail {

template <>
struct type_caster<std::span<int>> {
public:
    PYBIND11_TYPE_CASTER(std::span<int>, _("std::span<int>"));

    // 从Python对象转换到std::span<int>
    bool load(py::handle src, bool convert) {
        // 校验输入是否为int类型的numpy数组
        if (!py::isinstance<py::array_t<int>>(src)) {
            return false;
        }

        py::array_t<int> arr = src.cast<py::array_t<int>>();
        // 确保数组内存是C风格连续的，否则无法安全映射到std::span
        if (!arr.is_c_contiguous()) {
            return false;
        }

        // 直接绑定numpy数组的内存到std::span，零拷贝
        auto buf = arr.request();
        value = std::span<int>(static_cast<int*>(buf.ptr), buf.size);
        return true;
    }

    // 可选：实现从std::span<int>到Python对象的转换（如果需要返回给Python）
    static py::handle cast(std::span<int> src, py::return_value_policy policy, py::handle parent) {
        // 创建不持有内存所有权的numpy数组（零拷贝返回）
        py::array_t<int> arr(src.size(), src.data(), parent);
        return arr.release();
    }
};

} // namespace detail
} // namespace pybind11

绑定C++函数示例 #

PYBIND11_MODULE(your_module_name, m) {
    m.def("data", [](std::span<int> x) {
        return x.size();
    }, py::arg("x"));
}

关键注意事项 #

• 零拷贝的前提是numpy数组为连续内存（C风格），如果输入数组是非连续的，load函数会返回false，pybind11会尝试其他转换方式（比如自动转为std::vector<int>，但会产生拷贝）
• 转换过程中不会复制数组数据，std::span仅作为numpy内存的视图存在，需确保Python数组在C++函数执行期间不被销毁或修改

内容的提问来源于stack exchange，提问作者Contango