提交 1a6e03c6 authored 作者: Frederic's avatar Frederic 提交者: Arnaud Bergeron

Fix dtype for Elemwise.perform with multiple output

上级 350edca0
...@@ -3545,8 +3545,11 @@ class Composite(ScalarOp): ...@@ -3545,8 +3545,11 @@ class Composite(ScalarOp):
def impl(self, *inputs): def impl(self, *inputs):
output_storage = [[None] for i in xrange(self.nout)] output_storage = [[None] for i in xrange(self.nout)]
self.perform(None, inputs, output_storage) self.perform(None, inputs, output_storage)
return utils.to_return_values([storage[0] for storage in ret = utils.to_return_values([storage[0] for storage in
output_storage]) output_storage])
if self.nout > 1:
ret = tuple(ret)
return ret
def grad(self, inputs, output_grads): def grad(self, inputs, output_grads):
raise NotImplementedError("grad is not implemented for Composite") raise NotImplementedError("grad is not implemented for Composite")
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论