1 //===================================================== 2 // Copyright (C) 2014 Benoit Steiner <benoit.steiner.goog@gmail.com> 3 //===================================================== 4 // 5 // This Source Code Form is subject to the terms of the Mozilla 6 // Public License v. 2.0. If a copy of the MPL was not distributed 7 // with this file, You can obtain one at http://mozilla.org/MPL/2.0/. 8 // 9 #ifndef TENSOR_INTERFACE_HH 10 #define TENSOR_INTERFACE_HH 11 12 #include <unsupported/Eigen/CXX11/Tensor> 13 #include <vector> 14 #include "btl.hh" 15 16 using namespace Eigen; 17 18 template<class real> 19 class tensor_interface 20 { 21 public : 22 typedef real real_type; 23 typedef typename Eigen::Tensor<real,2>::Index Index; 24 25 typedef std::vector<real> stl_vector; 26 typedef std::vector<stl_vector> stl_matrix; 27 28 typedef Eigen::Tensor<real,2> gene_matrix; 29 typedef Eigen::Tensor<real,1> gene_vector; 30 31 name(void)32 static inline std::string name( void ) 33 { 34 return EIGEN_MAKESTRING(BTL_PREFIX); 35 } 36 free_matrix(gene_matrix &,int)37 static void free_matrix(gene_matrix & /*A*/, int /*N*/) {} 38 free_vector(gene_vector &)39 static void free_vector(gene_vector & /*B*/) {} 40 matrix_from_stl(gene_matrix & A,stl_matrix & A_stl)41 static BTL_DONT_INLINE void matrix_from_stl(gene_matrix & A, stl_matrix & A_stl){ 42 A.resize(Eigen::array<Index,2>(A_stl[0].size(), A_stl.size())); 43 44 for (unsigned int j=0; j<A_stl.size() ; j++){ 45 for (unsigned int i=0; i<A_stl[j].size() ; i++){ 46 A.coeffRef(Eigen::array<Index,2>(i,j)) = A_stl[j][i]; 47 } 48 } 49 } 50 vector_from_stl(gene_vector & B,stl_vector & B_stl)51 static BTL_DONT_INLINE void vector_from_stl(gene_vector & B, stl_vector & B_stl){ 52 B.resize(B_stl.size()); 53 54 for (unsigned int i=0; i<B_stl.size() ; i++){ 55 B.coeffRef(i) = B_stl[i]; 56 } 57 } 58 vector_to_stl(gene_vector & B,stl_vector & B_stl)59 static BTL_DONT_INLINE void vector_to_stl(gene_vector & B, stl_vector & B_stl){ 60 for (unsigned int i=0; i<B_stl.size() ; i++){ 61 B_stl[i] = B.coeff(i); 62 } 63 } 64 matrix_to_stl(gene_matrix & A,stl_matrix & A_stl)65 static BTL_DONT_INLINE void matrix_to_stl(gene_matrix & A, stl_matrix & A_stl){ 66 int N=A_stl.size(); 67 68 for (int j=0;j<N;j++){ 69 A_stl[j].resize(N); 70 for (int i=0;i<N;i++){ 71 A_stl[j][i] = A.coeff(Eigen::array<Index,2>(i,j)); 72 } 73 } 74 } 75 matrix_matrix_product(const gene_matrix & A,const gene_matrix & B,gene_matrix & X,int)76 static inline void matrix_matrix_product(const gene_matrix & A, const gene_matrix & B, gene_matrix & X, int /*N*/){ 77 typedef typename Eigen::Tensor<real_type, 1>::DimensionPair DimPair; 78 const Eigen::array<DimPair, 1> dims(DimPair(1, 0)); 79 X/*.noalias()*/ = A.contract(B, dims); 80 } 81 matrix_vector_product(const gene_matrix & A,const gene_vector & B,gene_vector & X,int)82 static inline void matrix_vector_product(const gene_matrix & A, const gene_vector & B, gene_vector & X, int /*N*/){ 83 typedef typename Eigen::Tensor<real_type, 1>::DimensionPair DimPair; 84 const Eigen::array<DimPair, 1> dims(DimPair(1, 0)); 85 X/*.noalias()*/ = A.contract(B, dims); 86 } 87 axpy(real coef,const gene_vector & X,gene_vector & Y,int)88 static inline void axpy(real coef, const gene_vector & X, gene_vector & Y, int /*N*/){ 89 Y += X.constant(coef) * X; 90 } 91 axpby(real a,const gene_vector & X,real b,gene_vector & Y,int)92 static inline void axpby(real a, const gene_vector & X, real b, gene_vector & Y, int /*N*/){ 93 Y = X.constant(a)*X + Y.constant(b)*Y; 94 } 95 copy_matrix(const gene_matrix & source,gene_matrix & cible,int)96 static EIGEN_DONT_INLINE void copy_matrix(const gene_matrix & source, gene_matrix & cible, int /*N*/){ 97 cible = source; 98 } 99 copy_vector(const gene_vector & source,gene_vector & cible,int)100 static EIGEN_DONT_INLINE void copy_vector(const gene_vector & source, gene_vector & cible, int /*N*/){ 101 cible = source; 102 } 103 }; 104 105 #endif 106