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

Convert scan output if needed

上级 05d008c7
......@@ -680,9 +680,12 @@ class Scan(theano.Op):
inplace_map)
# write to storage
# write to storage, converting if needed
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 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论