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

Adjust tests for gamma.

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