From 407b0efc795e84c9205d1748949a6eee2be7692c Mon Sep 17 00:00:00 2001
From: qijun <qijun1994@hotmail.com>
Date: Thu, 20 Jul 2017 11:10:31 +0800
Subject: [PATCH] fix bug in EigenVector::Flatten method

---
 paddle/framework/eigen.h | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/paddle/framework/eigen.h b/paddle/framework/eigen.h
index 4ba4fd4d110..f03b26a5fe4 100644
--- a/paddle/framework/eigen.h
+++ b/paddle/framework/eigen.h
@@ -65,13 +65,15 @@ template <typename T, int MajorType = Eigen::RowMajor,
           typename IndexType = Eigen::DenseIndex>
 struct EigenVector : public EigenTensor<T, 1, MajorType, IndexType> {
   // Flatten is to reshape a Tensor into a one dimension EigenVector
-  static typename EigenTensor<T, 1>::Type Flatten(Tensor& tensor) {
-    return EigenTensor<T, 1>::From(
+  static typename EigenTensor<T, 1, MajorType, IndexType>::Type Flatten(
+      Tensor& tensor) {
+    return EigenTensor<T, 1, MajorType, IndexType>::From(
         tensor, make_ddim({static_cast<int>(product(tensor.dims_))}));
   }
 
-  static typename EigenTensor<T, 1>::ConstType Flatten(const Tensor& tensor) {
-    return EigenTensor<T, 1>::From(
+  static typename EigenTensor<T, 1, MajorType, IndexType>::ConstType Flatten(
+      const Tensor& tensor) {
+    return EigenTensor<T, 1, MajorType, IndexType>::From(
         tensor, make_ddim({static_cast<int>(product(tensor.dims_))}));
   }
 };
-- 
GitLab