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

fix using disconnected as mem buffer

上级 fffe37c0
...@@ -1624,7 +1624,11 @@ class Scan(PureOp): ...@@ -1624,7 +1624,11 @@ class Scan(PureOp):
n_mitmot_inps = 0 n_mitmot_inps = 0
for idx in xrange(self.n_mit_mot): for idx in xrange(self.n_mit_mot):
outer_inp_mitmot.append(dC_douts[idx][::-1]) if isinstance(dC_douts[idx].type, DisconnectedType):
out = outs[idx]
outer_inp_mitmot.append(tensor.zeros_like(out))
else:
outer_inp_mitmot.append(dC_douts[idx][::-1])
mitmot_inp_taps.append([]) mitmot_inp_taps.append([])
mitmot_out_taps.append([]) mitmot_out_taps.append([])
undefined = False undefined = False
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论