pybind/pybind11
1
0
Fork 0
mirror of https://github.com/pybind/pybind11.git synced 2026-06-06 06:11:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

ability to prevent force casts in numpy arguments

Browse Source
This commit is contained in:
Wenzel Jakob
2016-05-19 16:02:09 +02:00
parent 93a317eca1
commit b47a9de035
5 changed files with 28 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
example/example10.cpp
View File

@@ -33,4 +33,9 @@ void init_ex10(py::module &m) {
// Vectorize a complex-valued function
m.def("vectorized_func3", py::vectorize(my_func3));
/// Numpy function which only accepts specific data types
m.def("selective_func", [](py::array_t<int, py::array::c_style>) { std::cout << "Int branch taken. "<< std::endl; });
m.def("selective_func", [](py::array_t<float, py::array::c_style>) { std::cout << "Float branch taken. "<< std::endl; });
m.def("selective_func", [](py::array_t<std::complex<float>, py::array::c_style>) { std::cout << "Complex float branch taken. "<< std::endl; });
}