提交 9c9d5036 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Convert scan output if needed

上级 05d008c7
...@@ -680,9 +680,12 @@ class Scan(theano.Op): ...@@ -680,9 +680,12 @@ class Scan(theano.Op):
inplace_map) inplace_map)
# write to storage # write to storage, converting if needed
for i in xrange(self.n_outs): for i in xrange(self.n_outs):
outs[i][0]=y[i] if hasattr(node.outputs[i], 'dtype'):
outs[i][0] = theano._asarray(y[i], dtype=node.outputs[i].dtype)
else:
outs[i][0] = y[i]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论