提交 73e5b801 authored 作者: Nicolas Bouchard's avatar Nicolas Bouchard

Adjust tests for gamma.

上级 4141a359
...@@ -1291,12 +1291,18 @@ _good_broadcast_unary_gammaln = dict( ...@@ -1291,12 +1291,18 @@ _good_broadcast_unary_gammaln = dict(
_grad_broadcast_unary_gammaln = dict( _grad_broadcast_unary_gammaln = dict(
normal=(rand_ranged(1e-8, 10, (2, 3)),),) normal=(rand_ranged(1e-8, 10, (2, 3)),),)
if theano.config.floatX == 'float32':
gamma_eps = 2e-4
else:
gamma_eps = 2e-10
GammaTester = makeBroadcastTester( GammaTester = makeBroadcastTester(
op=tensor.gamma, op=tensor.gamma,
expected=scipy.special.gamma, expected=scipy.special.gamma,
good=_good_broadcast_unary_gammaln, good=_good_broadcast_unary_gammaln,
grad=_grad_broadcast_unary_gammaln, grad=_grad_broadcast_unary_gammaln,
mode=mode_no_scipy, mode=mode_no_scipy,
eps=gamma_eps,
skip=skip_scipy) skip=skip_scipy)
GammaInplaceTester = makeBroadcastTester( GammaInplaceTester = makeBroadcastTester(
op=inplace.gamma_inplace, op=inplace.gamma_inplace,
...@@ -1304,6 +1310,7 @@ GammaInplaceTester = makeBroadcastTester( ...@@ -1304,6 +1310,7 @@ GammaInplaceTester = makeBroadcastTester(
good=_good_broadcast_unary_gammaln, good=_good_broadcast_unary_gammaln,
grad=_grad_broadcast_unary_gammaln, grad=_grad_broadcast_unary_gammaln,
mode=mode_no_scipy, mode=mode_no_scipy,
eps=gamma_eps,
inplace=True, inplace=True,
skip=skip_scipy) skip=skip_scipy)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论