From 2bb8f5b1b11dcce66c6512c0c67a7575de20a5d0 Mon Sep 17 00:00:00 2001
From: qingqing01 <dangqingqing@baidu.com>
Date: Wed, 19 Sep 2018 20:40:08 +0800
Subject: [PATCH] Support variable data shape of non-sequence data. (#13482)

---
 python/paddle/fluid/data_feeder.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/paddle/fluid/data_feeder.py b/python/paddle/fluid/data_feeder.py
index 631bbfe1fe..5102a558fd 100644
--- a/python/paddle/fluid/data_feeder.py
+++ b/python/paddle/fluid/data_feeder.py
@@ -71,7 +71,7 @@ class DataToLoDTensorConverter(object):
 
     def done(self):
         arr = numpy.array(self.data, dtype=self.dtype)
-        if self.shape:
+        if self.shape and len(arr.shape) != len(self.shape):
             arr = arr.reshape(self.shape)
         t = core.LoDTensor()
         t.set(arr, self.place)
-- 
GitLab