提交 e3c4c093 authored 作者: Frederic's avatar Frederic

Fix test by using the good cmp method.

上级 a09ba022
...@@ -67,8 +67,8 @@ class TestBatchedDot(TestCase): ...@@ -67,8 +67,8 @@ class TestBatchedDot(TestCase):
z_test = numpy.sum(a[:,:,:,None]*b[:,None,:,:],axis=-2) z_test = numpy.sum(a[:,:,:,None]*b[:,None,:,:],axis=-2)
assert numpy.allclose(z0, z_test) unittest_tools.assert_allclose(z0, z_test)
assert numpy.allclose(z1, z_test) unittest_tools.assert_allclose(z1, z_test)
cmp((5,4,3), (5,3,2)) cmp((5,4,3), (5,3,2))
cmp((5,3,3), (5,3,3)) cmp((5,3,3), (5,3,3))
...@@ -108,7 +108,7 @@ class TestBatchedDot(TestCase): ...@@ -108,7 +108,7 @@ class TestBatchedDot(TestCase):
self.assertRaises(RuntimeError, fail, (5,4,3), (5,2,2)) self.assertRaises(RuntimeError, fail, (5,4,3), (5,2,2))
def test_batched_dot_gradient(self): def test_batched_dot_gradient(self):
theano.tests.unittest_tools.verify_grad( unittest_tools.verify_grad(
batched_dot, batched_dot,
[numpy.random.randn(5,7,2).astype(numpy.float32), [numpy.random.randn(5,7,2).astype(numpy.float32),
numpy.random.randn(5,2,6).astype(numpy.float32)], numpy.random.randn(5,2,6).astype(numpy.float32)],
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论