提交 5df42b9c authored 作者: Razvan Pascanu's avatar Razvan Pascanu

fix inputs bug in scan

上级 246b7287
...@@ -704,11 +704,17 @@ def scan(fn, sequences=[], outputs_info=[], non_sequences=[], ...@@ -704,11 +704,17 @@ def scan(fn, sequences=[], outputs_info=[], non_sequences=[],
# make the compilation as fast as possible by not applying any optimization # make the compilation as fast as possible by not applying any optimization
# or conversion to C [ note this region is not important for performance # or conversion to C [ note this region is not important for performance
# so we can do stuff as unoptimal as we wish ] # so we can do stuff as unoptimal as we wish ]
'''
Why did I use gof.graph.inputs to pick the inputs here ??
dummy_f = function(filter(lambda x: isinstance(x,gof.Variable) and \ dummy_f = function(filter(lambda x: isinstance(x,gof.Variable) and \
not isinstance(x,SharedVariable) and not isinstance(x,gof.Constant), \ not isinstance(x,SharedVariable) and not isinstance(x,gof.Constant), \
reversed(gof.graph.inputs(dummy_args))), outputs, updates = updates, mode = \ reversed(gof.graph.inputs(dummy_args))), outputs, updates = updates, mode = \
compile.mode.Mode(linker = 'py', optimizer = None) ) compile.mode.Mode(linker = 'py', optimizer = None) )
'''
dummy_f = function(filter(lambda x: isinstance(x, gof.Variable) and \
not isinstance(x,SharedVariable) and not isinstance(x,gof.Constant), \
dummy_args), outputs, updates = updates, mode = compile.mode.Mode(linker='py',optimizer=None))
# We now look at what outputs our function returns # We now look at what outputs our function returns
inner_fn_outs = [ out.variable for out in dummy_f.maker.outputs] inner_fn_outs = [ out.variable for out in dummy_f.maker.outputs]
update_map = {} update_map = {}
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论