Support variable data shape of non-sequence data. (#13482)

7 years ago · 2bb8f5b1b1
parent 8a8c57267a
commit 2bb8f5b1b1
1 changed files with 1 additions and 1 deletions
--- 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)