From b5723823fff6127eda267f9f47fd158162fc9545 Mon Sep 17 00:00:00 2001
From: Justin Carpentier <justin.carpentier@inria.fr>
Date: Sun, 19 Feb 2023 19:24:26 +0100
Subject: [PATCH] core: add allocator for Eigen::Tensor
---
include/eigenpy/numpy-allocator.hpp | 21 +++++++++++++++++++--
1 file changed, 19 insertions(+), 2 deletions(-)
diff --git a/include/eigenpy/numpy-allocator.hpp b/include/eigenpy/numpy-allocator.hpp
index 30c28736..0a522550 100644
--- a/include/eigenpy/numpy-allocator.hpp
+++ b/include/eigenpy/numpy-allocator.hpp
@@ -1,12 +1,12 @@
/*
- * Copyright 2020-2022 INRIA
+ * Copyright 2020-2023 INRIA
*/
#ifndef __eigenpy_numpy_allocator_hpp__
#define __eigenpy_numpy_allocator_hpp__
-#include "eigenpy/eigen-allocator.hpp"
#include "eigenpy/fwd.hpp"
+#include "eigenpy/eigen-allocator.hpp"
#include "eigenpy/numpy-type.hpp"
#include "eigenpy/register.hpp"
@@ -29,6 +29,23 @@ struct NumpyAllocator {
return pyArray;
}
+
+#ifdef EIGENPY_WITH_TENSOR_SUPPORT
+ template <typename Scalar, int Rank, int Options, typename IndexType>
+ static PyArrayObject *allocate(
+ const Eigen::Tensor<Scalar, Rank, Options, IndexType> &tensor,
+ npy_intp nd, npy_intp *shape) {
+ typedef Eigen::Tensor<Scalar, Rank, Options, IndexType> Tensor;
+ const int code = Register::getTypeCode<Scalar>();
+ PyArrayObject *pyArray = (PyArrayObject *)call_PyArray_SimpleNew(
+ static_cast<int>(nd), shape, code);
+
+ // Copy data
+ EigenAllocator<Tensor>::copy(tensor, pyArray);
+
+ return pyArray;
+ }
+#endif
};
template <typename MatType>
--
GitLab