fix dim problem

8 years ago · 98bb8ee295
parent 633082ad9d
commit 98bb8ee295
1 changed files with 4 additions and 1 deletions
--- a/paddle/py_paddle/dataprovider_converter.py
+++ b/paddle/py_paddle/dataprovider_converter.py
@ -109,6 +109,9 @@ class DenseScanner(IScanner):
            if len(self.__shape__) > 3:
                raise ValueError(
                    "The dimension of input cannot be greater than 3.")
+            if len(self.__shape__) == 0:
+                self.__dim__ = 1
+            else:
                self.__dim__ = reduce(lambda x, y: x * y, self.__shape__)
            if len(self.__shape__) == 1 and self.__dim__ != self.input_type.dim:
                raise ValueError(