提交 8b69c134 编写于 作者: Q qiaolongfei

check shape of vector input, should not be a scalar

上级 98bb8ee2
...@@ -110,9 +110,9 @@ class DenseScanner(IScanner): ...@@ -110,9 +110,9 @@ class DenseScanner(IScanner):
raise ValueError( raise ValueError(
"The dimension of input cannot be greater than 3.") "The dimension of input cannot be greater than 3.")
if len(self.__shape__) == 0: if len(self.__shape__) == 0:
self.__dim__ = 1 raise ValueError(
else: "The input should be a vector, please check your input data.")
self.__dim__ = reduce(lambda x, y: x * y, self.__shape__) 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.")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册