提交 378430eb authored 作者: Ian Goodfellow's avatar Ian Goodfellow

fixed true_div gradient

上级 fbb5268e
...@@ -1325,7 +1325,7 @@ class TrueDiv(BinaryScalarOp): ...@@ -1325,7 +1325,7 @@ class TrueDiv(BinaryScalarOp):
if y.type in complex_types: if y.type in complex_types:
raise NotImplementedError() raise NotImplementedError()
second_part = cast(-(gz * x) / (y * y), y.type.dtype) second_part = -(gz * x) / (y * y)
return first_part, second_part return first_part, second_part
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论