提交 6dcc4f3c authored 作者: Frederic's avatar Frederic 提交者: Olivier Delalleau

Fix crash in new check in the scan op.

上级 7624f7d6
...@@ -1000,7 +1000,7 @@ class Scan(PureOp): ...@@ -1000,7 +1000,7 @@ class Scan(PureOp):
# is the shape of self.inputs[i] # is the shape of self.inputs[i]
for inp, inp_shp in izip(node.inputs, input_shapes): for inp, inp_shp in izip(node.inputs, input_shapes):
assert inp_shp is None or len(inp_shp) == inp.ndim assert inp_shp is None or len(inp_shp) == inp.type.ndim
# sequences # sequences
# We skip iputs_shapes[0] as it is the total or current number # We skip iputs_shapes[0] as it is the total or current number
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论