提交 2681cd70 authored 作者: Gijs van Tulder's avatar Gijs van Tulder

Use utt.assert_allclose for more informative CorrMM tests failures.

上级 99cca895
...@@ -10,7 +10,6 @@ import theano ...@@ -10,7 +10,6 @@ import theano
import theano.tensor as T import theano.tensor as T
from theano.tests import unittest_tools as utt from theano.tests import unittest_tools as utt
from theano.tensor.nnet import corr, conv from theano.tensor.nnet import corr, conv
from theano.tensor.basic import _allclose
class TestCorr2D(utt.InferShapeTester): class TestCorr2D(utt.InferShapeTester):
...@@ -132,7 +131,7 @@ class TestCorr2D(utt.InferShapeTester): ...@@ -132,7 +131,7 @@ class TestCorr2D(utt.InferShapeTester):
icol:icol + dil_fil_shape2d[1]:filter_dilation[1]] * filter2d[::-1, ::-1] icol:icol + dil_fil_shape2d[1]:filter_dilation[1]] * filter2d[::-1, ::-1]
).sum() ).sum()
self.assertTrue(_allclose(theano_output, ref_output)) utt.assert_allclose(theano_output, ref_output)
# TEST GRADIENT # TEST GRADIENT
if verify_grad: if verify_grad:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论