提交 610952f2 authored 作者: Razvan Pascanu's avatar Razvan Pascanu

small fix to scan

上级 e8756c30
...@@ -1572,7 +1572,9 @@ class Scan(Op): ...@@ -1572,7 +1572,9 @@ class Scan(Op):
self.go_backwards, self.seqs_taps, self.outs_taps, self.go_backwards, self.seqs_taps, self.outs_taps,
truncate_gradient) truncate_gradient)
g_scan_outs = g_scan(g_args) g_scan_outs = g_scan(g_args)
# We need to add several None's fpr shared vars with updates if not type(g_scan_outs) in (list, tuple):
g_scan_outs = [ g_scan_outs ]
# We need to add several None's for shared vars with updates
gradients = [None] + g_scan_outs[:self.n_seqs+self.n_outs_not_shared] gradients = [None] + g_scan_outs[:self.n_seqs+self.n_outs_not_shared]
gradients += [None for i in xrange(self.n_outs-self.n_outs_not_shared)] gradients += [None for i in xrange(self.n_outs-self.n_outs_not_shared)]
gradients += g_scan_outs[self.n_seqs+self.n_outs_not_shared:] gradients += g_scan_outs[self.n_seqs+self.n_outs_not_shared:]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论