如何通过pybind11使用C++中的numpy数组？

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

namespace py = pybind11;

void process_numpy_array(py::array_t<double> input_array) {
    // 在这里对numpy数组进行操作
    // 例如，可以使用py::array_t::mutable_data()函数获取数组的指针，并对数组进行修改
    double* data = input_array.mutable_data();
    // ...
}

PYBIND11_MODULE(example, m) {
    m.def("process_numpy_array", &process_numpy_array, "Process numpy array in C++");
}

import numpy as np
import example

# 创建一个numpy数组
arr = np.array([1.0, 2.0, 3.0])

# 调用C++函数处理numpy数组
example.process_numpy_array(arr)