fix dim problem

gangliao-patch-1
qiaolongfei 8 years ago
parent 633082ad9d
commit 98bb8ee295

@ -109,7 +109,10 @@ class DenseScanner(IScanner):
if len(self.__shape__) > 3: if len(self.__shape__) > 3:
raise ValueError( raise ValueError(
"The dimension of input cannot be greater than 3.") "The dimension of input cannot be greater than 3.")
self.__dim__ = reduce(lambda x, y: x * y, self.__shape__) 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: if len(self.__shape__) == 1 and self.__dim__ != self.input_type.dim:
raise ValueError( raise ValueError(
"The data size must be equal to it in data layer.") "The data size must be equal to it in data layer.")

Loading…
Cancel
Save