提交 50f178c9 authored 作者: Razvan Pascanu's avatar Razvan Pascanu

add missing arguments to the scan op

上级 f4b40293
...@@ -492,6 +492,8 @@ def scan(fn, ...@@ -492,6 +492,8 @@ def scan(fn,
outputs=outputs, outputs=outputs,
lengths=lengths, lengths=lengths,
switches=[], switches=[],
mintaps=mintaps,
index=t,
options=options, options=options,
as_repeatUntil=cond) as_repeatUntil=cond)
# Note that we get here all the outputs followed by the update rules to # Note that we get here all the outputs followed by the update rules to
......
...@@ -45,11 +45,14 @@ class ScanOp(PureOp): ...@@ -45,11 +45,14 @@ class ScanOp(PureOp):
outputs, outputs,
lengths, lengths,
switches, switches,
mintaps,
index,
options, options,
as_repeatUntil): as_repeatUntil):
self.inputs = inputs self.inputs = inputs
self.outputs = outputs self.outputs = outputs
self.index = index
self.switches = switches self.switches = switches
self.lengths = lengths self.lengths = lengths
self.mintaps = mintaps self.mintaps = mintaps
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论