提交 5d8a7ae8 authored 作者: Razvan Pascanu's avatar Razvan Pascanu

add None as n_steps only if there is at least one sequence

上级 fd52133e
...@@ -228,7 +228,9 @@ class TestScan(unittest.TestCase): ...@@ -228,7 +228,9 @@ class TestScan(unittest.TestCase):
for out in out_mem_buffers: for out in out_mem_buffers:
out[step] = 2 out[step] = 2
return nw_states_outs + out_mem_buffers, shared_values return nw_states_outs + out_mem_buffers, shared_values
possible_n_steps = [-1, 1, 5, -5]
if n_ins > 0:
possible_n_steps.append(None)
for n_steps in [-1, 1, 5, -5, None]: for n_steps in [-1, 1, 5, -5, None]:
for go_backwards in [True, False]: for go_backwards in [True, False]:
outputs, updates = scan_module.scan( outputs, updates = scan_module.scan(
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论