generate more compact binaries

example/example.cpp

@@ -18,6 +18,7 @@ void init_ex6(py::module &);
 void init_ex7(py::module &);
 void init_ex8(py::module &);
 void init_ex9(py::module &);
+void init_ex10(py::module &);
 
 PYTHON_PLUGIN(example) {
     py::module m("example", "pybind example plugin");
@@ -31,6 +32,7 @@ PYTHON_PLUGIN(example) {
     init_ex7(m);
     init_ex8(m);
     init_ex9(m);
+    init_ex10(m);
 
     return m.ptr();
 }

example/example10.cpp

@@ -0,0 +1,28 @@
+/*
+    example/example10.cpp -- Example 10: auto-vectorize functions for NumPy
+
+    Copyright (c) 2015 Wenzel Jakob <wenzel@inf.ethz.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 <pybind/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 x*y*z;
+}
+
+void init_ex10(py::module &m) {
+    // Vectorize all arguments (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_dtype<int> x, py::array_dtype<float> y, float z) {
+            return py::vectorize([z](int x, float y) { return my_func(x, y, z); })(x, y);
+        }
+    );
+}

example/example10.py

@@ -0,0 +1,16 @@
+#!/usr/bin/env python3
+import sys
+sys.path.append('.')
+
+import example
+import numpy as np
+
+from example import vectorized_func
+from example import vectorized_func2
+
+for f in [vectorized_func, vectorized_func2]:
+    print(f(1, 2, 3))
+    print(f(np.array(1), np.array(2), 3))
+    print(f(np.array([1, 3]), np.array([2, 4]), 3))
+    print(f(np.array([[1, 3, 5], [7, 9, 11]]), np.array([[2, 4, 6], [8, 10, 12]]), 3))
+    print(np.array([[1, 3, 5], [7, 9, 11]])* np.array([[2, 4, 6], [8, 10, 12]])*3)