mirror of
https://github.com/pybind/pybind11.git
synced 2024-11-30 00:47:12 +00:00
b3f3d79f4c
This renames example files from `exampleN` to `example-description`.
Specifically, the following renaming is applied:
example1 -> example-methods-and-attributes
example2 -> example-python-types
example3 -> example-operator-overloading
example4 -> example-constants-and-functions
example5 -> example-callbacks (*)
example6 -> example-sequence-and-iterators
example7 -> example-buffers
example8 -> example-custom-ref-counting
example9 -> example-modules
example10 -> example-numpy-vectorize
example11 -> example-arg-keywords-and-defaults
example12 -> example-virtual-functions
example13 -> example-keep-alive
example14 -> example-opaque-types
example15 -> example-pickling
example16 -> example-inheritance
example17 -> example-stl-binders
example18 -> example-eval
example19 -> example-custom-exceptions
* the inheritance parts of example5 are moved into example-inheritance
(previously example16), and the remainder is left as example-callbacks.
This commit also renames the internal variables ("Example1",
"Example2", "Example4", etc.) into non-numeric names ("ExampleMandA",
"ExamplePythonTypes", "ExampleWithEnum", etc.) to correspond to the
file renaming.
The order of tests is preserved, but this can easily be changed if
there is some more natural ordering by updating the list in
examples/CMakeLists.txt.
42 lines
1.6 KiB
C++
42 lines
1.6 KiB
C++
/*
|
|
example/example-numpy-vectorize.cpp -- auto-vectorize functions over NumPy array
|
|
arguments
|
|
|
|
Copyright (c) 2016 Wenzel Jakob <wenzel.jakob@epfl.ch>
|
|
|
|
All rights reserved. Use of this source code is governed by a
|
|
BSD-style license that can be found in the LICENSE file.
|
|
*/
|
|
|
|
#include "example.h"
|
|
#include <pybind11/numpy.h>
|
|
|
|
double my_func(int x, float y, double z) {
|
|
std::cout << "my_func(x:int=" << x << ", y:float=" << y << ", z:float=" << z << ")" << std::endl;
|
|
return (float) x*y*z;
|
|
}
|
|
|
|
std::complex<double> my_func3(std::complex<double> c) {
|
|
return c * std::complex<double>(2.f);
|
|
}
|
|
|
|
void init_ex_numpy_vectorize(py::module &m) {
|
|
// Vectorize all arguments of a function (though non-vector arguments are also allowed)
|
|
m.def("vectorized_func", py::vectorize(my_func));
|
|
|
|
// Vectorize a lambda function with a capture object (e.g. to exclude some arguments from the vectorization)
|
|
m.def("vectorized_func2",
|
|
[](py::array_t<int> x, py::array_t<float> y, float z) {
|
|
return py::vectorize([z](int x, float y) { return my_func(x, y, z); })(x, y);
|
|
}
|
|
);
|
|
|
|
// 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; });
|
|
}
|