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

fix bugs for arctanh

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