提交 cd4c269c authored 作者: Frederic Bastien's avatar Frederic Bastien

use numpy.allclose instead of numpy.all in test.

上级 a22ff6eb
...@@ -2471,7 +2471,7 @@ class test_tensordot(unittest.TestCase): ...@@ -2471,7 +2471,7 @@ class test_tensordot(unittest.TestCase):
f2 = inplace_func([avec,bmat],c) f2 = inplace_func([avec,bmat],c)
aval = self.rand(5); aval = self.rand(5);
bval = self.rand(8,5); bval = self.rand(8,5);
self.failUnless(numpy.all(numpy.tensordot(aval,bval,axes) == \ self.failUnless(numpy.allclose(numpy.tensordot(aval,bval,axes),
f2(aval,bval))) f2(aval,bval)))
utt.verify_grad(TensorDot(axes), [aval,bval]) utt.verify_grad(TensorDot(axes), [aval,bval])
...@@ -2482,7 +2482,7 @@ class test_tensordot(unittest.TestCase): ...@@ -2482,7 +2482,7 @@ class test_tensordot(unittest.TestCase):
f3 = inplace_func([amat,bmat],c) f3 = inplace_func([amat,bmat],c)
aval = self.rand(4,7); aval = self.rand(4,7);
bval = self.rand(7,9); bval = self.rand(7,9);
self.failUnless(numpy.all(numpy.tensordot(aval,bval,axes) == \ self.failUnless(numpy.allclose(numpy.tensordot(aval,bval,axes),
f3(aval,bval))) f3(aval,bval)))
utt.verify_grad(TensorDot(axes), [aval,bval]) utt.verify_grad(TensorDot(axes), [aval,bval])
...@@ -2493,7 +2493,7 @@ class test_tensordot(unittest.TestCase): ...@@ -2493,7 +2493,7 @@ class test_tensordot(unittest.TestCase):
f4 = inplace_func([atens,bmat],c) f4 = inplace_func([atens,bmat],c)
aval = self.rand(1,2,3,4); aval = self.rand(1,2,3,4);
bval = self.rand(2,3); bval = self.rand(2,3);
self.failUnless(numpy.all(numpy.tensordot(aval,bval,axes) == \ self.failUnless(numpy.allclose(numpy.tensordot(aval,bval,axes),
f4(aval,bval))) f4(aval,bval)))
utt.verify_grad(TensorDot(axes), [aval,bval]) utt.verify_grad(TensorDot(axes), [aval,bval])
...@@ -2505,14 +2505,14 @@ class test_tensordot(unittest.TestCase): ...@@ -2505,14 +2505,14 @@ class test_tensordot(unittest.TestCase):
f5 = inplace_func([atens,btens],c) f5 = inplace_func([atens,btens],c)
aval = self.rand(4,3,5,2); aval = self.rand(4,3,5,2);
bval = self.rand(3,4,2); bval = self.rand(3,4,2);
self.failUnless(numpy.all(numpy.tensordot(aval,bval,axes) == \ self.failUnless(numpy.allclose(numpy.tensordot(aval,bval,axes),
f5(aval,bval))) f5(aval,bval)))
utt.verify_grad(TensorDot(axes), [aval,bval]) utt.verify_grad(TensorDot(axes), [aval,bval])
axes = (axes[1],axes[0]) axes = (axes[1],axes[0])
c = tensordot(axes)(btens, atens) c = tensordot(axes)(btens, atens)
f6 = inplace_func([btens,atens],c) f6 = inplace_func([btens,atens],c)
self.failUnless(numpy.all(numpy.tensordot(bval,aval,axes) == \ self.failUnless(numpy.allclose(numpy.tensordot(bval,aval,axes),
f6(bval,aval))) f6(bval,aval)))
utt.verify_grad(TensorDot(axes), [bval,aval]) utt.verify_grad(TensorDot(axes), [bval,aval])
...@@ -2554,7 +2554,7 @@ class test_tensordot(unittest.TestCase): ...@@ -2554,7 +2554,7 @@ class test_tensordot(unittest.TestCase):
f3 = inplace_func([amat,bmat],c) f3 = inplace_func([amat,bmat],c)
aval = self.rand(4,7); aval = self.rand(4,7);
bval = self.rand(7,9); bval = self.rand(7,9);
self.failUnless(numpy.all(numpy.tensordot(aval,bval,axes) == \ self.failUnless(numpy.allclose(numpy.tensordot(aval,bval,axes),
f3(aval,bval))) f3(aval,bval)))
utt.verify_grad(TensorDot(axes), [aval,bval]) utt.verify_grad(TensorDot(axes), [aval,bval])
...@@ -2567,7 +2567,7 @@ class test_tensordot(unittest.TestCase): ...@@ -2567,7 +2567,7 @@ class test_tensordot(unittest.TestCase):
bval = numpy.random.rand(5,3) bval = numpy.random.rand(5,3)
c = tensordot(axes)(amat, bmat) c = tensordot(axes)(amat, bmat)
f3 = inplace_func([amat,bmat],c) f3 = inplace_func([amat,bmat],c)
self.failUnless(numpy.all(numpy.tensordot(aval,bval,axes) == \ self.failUnless(numpy.allclose(numpy.tensordot(aval,bval,axes),
f3(aval,bval))) f3(aval,bval)))
utt.verify_grad(TensorDot(axes), [aval,bval]) utt.verify_grad(TensorDot(axes), [aval,bval])
...@@ -2579,7 +2579,7 @@ class test_tensordot(unittest.TestCase): ...@@ -2579,7 +2579,7 @@ class test_tensordot(unittest.TestCase):
bval = self.rand(4,5,3) bval = self.rand(4,5,3)
c = tensordot(axes)(amat, bmat) c = tensordot(axes)(amat, bmat)
f3 = inplace_func([amat,bmat],c) f3 = inplace_func([amat,bmat],c)
self.failUnless(numpy.all(numpy.tensordot(aval,bval,axes) == \ self.failUnless(numpy.allclose(numpy.tensordot(aval,bval,axes),
f3(aval,bval))) f3(aval,bval)))
utt.verify_grad(TensorDot(axes), [aval,bval]) utt.verify_grad(TensorDot(axes), [aval,bval])
...@@ -2592,7 +2592,7 @@ class test_tensordot(unittest.TestCase): ...@@ -2592,7 +2592,7 @@ class test_tensordot(unittest.TestCase):
bval = self.rand(5,4) bval = self.rand(5,4)
c = tensordot(axes)(amat, bmat) c = tensordot(axes)(amat, bmat)
f3 = inplace_func([amat,bmat],c) f3 = inplace_func([amat,bmat],c)
self.failUnless(numpy.all(numpy.tensordot(aval,bval,axes) == \ self.failUnless(numpy.allclose(numpy.tensordot(aval,bval,axes),
f3(aval,bval))) f3(aval,bval)))
utt.verify_grad(TensorDot(axes), [aval,bval]) utt.verify_grad(TensorDot(axes), [aval,bval])
...@@ -2610,8 +2610,8 @@ class test_tensordot(unittest.TestCase): ...@@ -2610,8 +2610,8 @@ class test_tensordot(unittest.TestCase):
f2 = inplace_func([amat,bmat,gzmat],tensordot_grad(((1,),(0,)))(amat, bmat, gzmat)) f2 = inplace_func([amat,bmat,gzmat],tensordot_grad(((1,),(0,)))(amat, bmat, gzmat))
o1=f1(aval,bval,gzval) o1=f1(aval,bval,gzval)
o2=f2(aval,bval,gzval) o2=f2(aval,bval,gzval)
self.failUnless(numpy.all(o1[0]==o2[0])) self.failUnless(numpy.allclose(o1[0],o2[0]))
self.failUnless(numpy.all(o1[1]==o2[1])) self.failUnless(numpy.allclose(o1[1],o2[1]))
def test_smallest_stack(): def test_smallest_stack():
sx, sy = dscalar(), dscalar() sx, sy = dscalar(), dscalar()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论