Zengtudor/pybind11
1
0
Fork 0
mirror of https://github.com/pybind/pybind11.git synced 2024-11-25 22:52:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

Eigen tests: '2*' functions for col-, row-vectors

Browse Source
This commit is contained in:
Ben North 2016-07-05 20:01:11 +01:00
parent 3b48482f02
commit b063e64b19
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
example/eigen.cpp
View File

@ -10,6 +10,12 @@
#include "example.h" #include "example.h"
#include <pybind11/eigen.h> #include <pybind11/eigen.h>
Eigen::VectorXf double_col(const Eigen::VectorXf& x)
{ return 2.0f * x; }
Eigen::RowVectorXf double_row(const Eigen::RowVectorXf& x)
{ return 2.0f * x; }
void init_eigen(py::module &m) { void init_eigen(py::module &m) {
typedef Eigen::Matrix<float, 5, 6, Eigen::RowMajor> FixedMatrixR; typedef Eigen::Matrix<float, 5, 6, Eigen::RowMajor> FixedMatrixR;
typedef Eigen::Matrix<float, 5, 6> FixedMatrixC; typedef Eigen::Matrix<float, 5, 6> FixedMatrixC;
@ -23,6 +29,9 @@ void init_eigen(py::module &m) {
mat << 0, 3, 0, 0, 0, 11, 22, 0, 0, 0, 17, 11, 7, 5, 0, 1, 0, 11, 0, mat << 0, 3, 0, 0, 0, 11, 22, 0, 0, 0, 17, 11, 7, 5, 0, 1, 0, 11, 0,
0, 0, 0, 0, 11, 0, 0, 14, 0, 8, 11; 0, 0, 0, 0, 11, 0, 0, 14, 0, 8, 11;
m.def("double_col", &double_col);
m.def("double_row", &double_row);
m.def("fixed_r", [mat]() -> FixedMatrixR { m.def("fixed_r", [mat]() -> FixedMatrixR {
return FixedMatrixR(mat); return FixedMatrixR(mat);
}); });