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

make the permutation after scan aware of the mintaps

上级 a0b4d6ef
...@@ -506,8 +506,11 @@ def scan(fn, ...@@ -506,8 +506,11 @@ def scan(fn,
scan_inputs = [T] + inputs scan_inputs = [T] + inputs
scan_outputs_update_rules = scan_utils.to_list(local_op(*scan_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 = [scan_utils.scan_permute(x, t) for x in scan_outputs] for pos in xrange(len(states_and_outputs)):
out = scan_utils.ScanPermutation(mintaps[pos])(
scan_outputs_update_rules[pos],t)
scan_outputs.append(out[mintap:])
# 5.6 Construct updates dictionary # 5.6 Construct updates dictionary
update_rules = scan_outputs_update_rules[len(states_and_outputs):] update_rules = scan_outputs_update_rules[len(states_and_outputs):]
updates = {} updates = {}
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论