提交 93978136 authored 作者: Razvan Pascanu's avatar Razvan Pascanu

fix infershape (the special case is if it behaves like a repeat until)

上级 3fa01756
......@@ -154,9 +154,8 @@ class ScanOp(PureOp):
def infer_shape(self, node, input_shapes):
for inp, inp_shp in izip(node.inputs, input_shapes):
assert inp_shp is None or len(inp_shp) == inp.type.ndim
n_outs = len(self.outputs)
if self.gpu:
if self.as_repeatUntil is not None:
return [(Shape_i(0)(o),) + x[1:] for o, x
in izip(node.outputs, input_shapes[1: n_outs + 1])]
else:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论