提交 43c14abb authored 作者: TimSalimans's avatar TimSalimans

Fix grad

上级 bdf60983
......@@ -112,7 +112,7 @@ class Erfcx(UnaryScalarOp):
cst = numpy.asarray(2. / numpy.sqrt(numpy.pi),
dtype=upcast(x.type.dtype, gz.type.dtype))
return - gz * cst + (2. * x) * erfcx(x),
return gz * (-cst + (2. * x) * erfcx(x)),
erfcx = Erfcx(upgrade_to_float_no_complex, name='erfcx')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论