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

add the number of steps to scan

上级 50f178c9
...@@ -503,7 +503,8 @@ def scan(fn, ...@@ -503,7 +503,8 @@ def scan(fn,
# * len(additional_input_states) updates for numeric shared variable # * len(additional_input_states) updates for numeric shared variable
# * len(non_numeric_input_states) updates for non numeric shared # * len(non_numeric_input_states) updates for non numeric shared
# variables # variables
scan_outputs_update_rules = local_op(*inputs) scan_inputs = [T] + inputs
scan_outputs_update_rules = scan_utils.to_list(local_op(*scan_inputs))
# 5.5 Collect outputs and add permutation object # 5.5 Collect outputs and add permutation object
scan_outputs = scan_outputs_update_rules[:len(states_and_outputs)] scan_outputs = scan_outputs_update_rules[:len(states_and_outputs)]
scan_outputs = [scan_utils.scan_permute(x, t) for x in scan_outputs] scan_outputs = [scan_utils.scan_permute(x, t) for x in scan_outputs]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论