如何使用pybind11绑定一个以numpy.array()为参数的函数，例如，shape (10，10，3)？

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

namespace py = pybind11;

double sum_array(py::array_t<double> input_array) {
    py::buffer_info buf_info = input_array.request();
    double* ptr = static_cast<double*>(buf_info.ptr);

    double sum = 0.0;
    for (size_t i = 0; i < buf_info.size; ++i) {
        sum += ptr[i];
    }

    return sum;
}

PYBIND11_MODULE(example, m) {
    m.def("sum_array", &sum_array, "Calculate the sum of an array");
}

import example
import numpy as np

arr = np.ones((10, 10, 3))
result = example.sum_array(arr)
print(result)