提交 38a8c842 authored 作者: Razvan Pascanu's avatar Razvan Pascanu

define n_steps for test

上级 f6fae119
......@@ -502,6 +502,7 @@ class TestScan(unittest.TestCase):
x0 = theano.tensor.scalar('x0')
W_in = theano.tensor.scalar('win')
W = theano.tensor.scalar('w')
n_steps = 5
output, updates = scan_module.scan(f_rnn,
u,
x0,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论