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

add a slice for every tap

上级 3f66e492
......@@ -565,7 +565,8 @@ def one_step_scan(fn,
arg_info)
# go through the taps
mintap = abs(numpy.min(arg_info['taps']))
states_slices.append(arg_info['initial'][k + mintap])
states_slices.extend(
[arg_info['initial'][k + mintap] for k in arg_info['taps']])
# Re-order args
args = (inputs_slices + states_slices + parameters)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论