提交 774c32ab authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Ricardo Vieira

Fix bug in gradient of Elemwise containing multi-output scalars

上级 a933dff9
......@@ -636,6 +636,9 @@ class Elemwise(OpenMPOp):
return DimShuffle((), ["x"] * nd)(res)
new_r = Elemwise(node.op, {})(*[transform(ipt) for ipt in node.inputs])
if isinstance(new_r, (list, tuple)):
# Scalar Op with multiple outputs
new_r = new_r[r.owner.outputs.index(r)]
return new_r
ret = []
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论