提交 a35ff4eb authored 作者: TimSalimans's avatar TimSalimans

Try smaller test range erfcx

上级 37944556
...@@ -1080,6 +1080,11 @@ _good_broadcast_unary_normal_float_no_empty_no_complex = copymod( ...@@ -1080,6 +1080,11 @@ _good_broadcast_unary_normal_float_no_empty_no_complex = copymod(
_good_broadcast_unary_normal_float_no_complex = copymod( _good_broadcast_unary_normal_float_no_complex = copymod(
_good_broadcast_unary_normal_float, _good_broadcast_unary_normal_float,
without=['complex']) without=['complex'])
_good_broadcast_unary_normal_float_no_complex_small_neg_range = dict(
normal=[rand_ranged(-2, 5, (2, 3))],
corner_case=[corner_case],
empty=[numpy.asarray([], dtype=config.floatX)])
_good_broadcast_unary_normal = dict( _good_broadcast_unary_normal = dict(
normal=[numpy.asarray(rand_ranged(-5, 5, (2, 3)), normal=[numpy.asarray(rand_ranged(-5, 5, (2, 3)),
...@@ -1109,6 +1114,10 @@ _grad_broadcast_unary_normal = dict( ...@@ -1109,6 +1114,10 @@ _grad_broadcast_unary_normal = dict(
corner_case=[corner_case_grad], corner_case=[corner_case_grad],
# empty = [numpy.asarray([])] # XXX: should this be included? # empty = [numpy.asarray([])] # XXX: should this be included?
) )
_grad_broadcast_unary_normal_small_neg_range = dict(
normal=[numpy.asarray(rand_ranged(-2, 5, (2, 3)), dtype=floatX)],
corner_case=[corner_case_grad])
_grad_broadcast_unary_normal_no_complex_no_corner_case = copymod( _grad_broadcast_unary_normal_no_complex_no_corner_case = copymod(
_grad_broadcast_unary_normal_no_complex, _grad_broadcast_unary_normal_no_complex,
...@@ -1709,16 +1718,16 @@ ErfcInplaceTester = makeBroadcastTester( ...@@ -1709,16 +1718,16 @@ ErfcInplaceTester = makeBroadcastTester(
ErfcxTester = makeBroadcastTester( ErfcxTester = makeBroadcastTester(
op=tensor.erfcx, op=tensor.erfcx,
expected=expected_erfcx, expected=expected_erfcx,
good=_good_broadcast_unary_normal_float_no_complex, good=_good_broadcast_unary_normal_float_no_complex_small_neg_range,
grad=_grad_broadcast_unary_normal, grad=_grad_broadcast_unary_normal_small_neg_range,
eps=2e-10, eps=2e-10,
mode=mode_no_scipy, mode=mode_no_scipy,
skip=skip_scipy12) skip=skip_scipy12)
ErfcxInplaceTester = makeBroadcastTester( ErfcxInplaceTester = makeBroadcastTester(
op=inplace.erfcx_inplace, op=inplace.erfcx_inplace,
expected=expected_erfcx, expected=expected_erfcx,
good=_good_broadcast_unary_normal_float_no_complex, good=_good_broadcast_unary_normal_float_no_complex_small_neg_range,
grad=_grad_broadcast_unary_normal, grad=_grad_broadcast_unary_normal_small_neg_range,
eps=2e-10, eps=2e-10,
mode=mode_no_scipy, mode=mode_no_scipy,
inplace=True, inplace=True,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论