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

better looking code

上级 0074516f
...@@ -463,8 +463,9 @@ def scan(fn, ...@@ -463,8 +463,9 @@ def scan(fn,
input_state = states_and_outputs_info[n]['initial'] input_state = states_and_outputs_info[n]['initial']
inputs.append(input_state) inputs.append(input_state)
outputs.append( outputs.append(
tensor.set_subtensor(input_state[(t + mintap) % lengths[n]], tensor.set_subtensor(
states_and_outputs[n])) input_state[(t + mintap) % lengths[n]],
states_and_outputs[n]))
else: else:
mem_buffer = scan_utils.allocate_memory( mem_buffer = scan_utils.allocate_memory(
T, states_and_outputs_info[n], states_and_outputs[n]) T, states_and_outputs_info[n], states_and_outputs[n])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论