提交 0fbd7e4c authored 作者: rman@rpad's avatar rman@rpad

small changes to scan op

上级 e1e700e1
......@@ -394,7 +394,7 @@ class Scan(theano.Op):
fn_args += list(args[(n_seqs+n_outs):])
# compute output
something = fn(*fn_args)
#update outputs
#update outputs
for j in xrange(n_outs):
y[j][i] = something[j]
return y
......@@ -406,7 +406,7 @@ class Scan(theano.Op):
warning('Can not compute gradients if inplace or updates ' \
'are used. Use force_gradient if you know for sure '\
'that the gradient can be computed automatically.')
return [None for i in inputs]
return [None for i in args]
else:
# forward pass
y = self(*args)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论