提交 b47331bd authored 作者: Nicolas Bouchard's avatar Nicolas Bouchard

fix bugs for arctanh

上级 61c52438
......@@ -2247,7 +2247,7 @@ class ArcTanh(UnaryScalarOp):
if x.type in complex_types:
raise NotImplementedError()
if x.type in float_types:
return gz / (numpy.cast[x.type](1) -sqr(x)),
return gz / (numpy.cast[x.type](1) - sqr(x)),
else:
return None,
......
......@@ -1203,11 +1203,14 @@ TanhInplaceTester = makeBroadcastTester(op=inplace.tanh_inplace,
grad=_grad_broadcast_unary_normal,
inplace=True)
_good_broadcast_unary_arctanh = dict(normal=(rand_ranged(-1, 1, (2, 3)),),
integers=(randint_ranged(-1, 1, (2, 3)),),
complex=(randc128_ranged(-1, 1, (2, 3)),),
empty=(numpy.asarray([]),),)
_grad_broadcast_unary_arctanh = dict(normal=(rand_ranged(-1, 1, (2, 3)),),)
_eps = 1e-10
_good_broadcast_unary_arctanh = dict(
normal=(rand_ranged(-1 + _eps, 1 - _eps, (2, 3)),),
integers=(randint_ranged(-1 + _eps, 1 - _eps, (2, 3)),),
complex=(randc128_ranged(-1 + _eps, 1 - _eps, (2, 3)),),
empty=(numpy.asarray([]),),)
_grad_broadcast_unary_arctanh = dict(
normal=(rand_ranged(-1 + _eps, 1 - _eps, (2, 3)),),)
ArcTanhTester = makeBroadcastTester(op=tensor.arctanh,
expected=numpy.arctanh,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论