提交 5467263a authored 作者: Razvan Pascanu's avatar Razvan Pascanu

attempt to fix he shared variables bit

上级 41ebc083
...@@ -455,6 +455,17 @@ def scan(fn, ...@@ -455,6 +455,17 @@ def scan(fn,
non_numeric_output_states.append(update_d[sv]) non_numeric_output_states.append(update_d[sv])
original_non_numeric_shared_variables.append(sv) original_non_numeric_shared_variables.append(sv)
# Replace shared variables in the update
_additional_output_states = []
replace = {}
for sv, buf in zip(original_numeric_shared_variables,
additional_input_states):
replace[sv] = buf[t]
for out in additional_output_states:
_additional_output_states.append(
scan_utils.clone(out, replace=replace))
additional_output_states = _additional_output_states
# 5.2 Collect inputs/outputs of the inner function # 5.2 Collect inputs/outputs of the inner function
inputs = [] inputs = []
outputs = [] outputs = []
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论