提交 086ee5b7 authored 作者: Frederic's avatar Frederic

Use all close in a test to make it pass more frequently.

上级 04aa2ef9
...@@ -5456,8 +5456,9 @@ class test_tensordot(unittest.TestCase): ...@@ -5456,8 +5456,9 @@ class test_tensordot(unittest.TestCase):
f1 = inplace_func([avec, bvec], c) f1 = inplace_func([avec, bvec], c)
aval = rand(5) aval = rand(5)
bval = rand(5) bval = rand(5)
self.assertTrue(numpy.tensordot(aval, bval, axes) == \ out0 = numpy.tensordot(aval, bval, axes)
f1(aval, bval)) out1 = f1(aval, bval)
self.assertTrue(numpy.allclose(out0, out1), (out0, out1))
utt.verify_grad(self.TensorDot(axes), [aval, bval]) utt.verify_grad(self.TensorDot(axes), [aval, bval])
# Test matrix-vector # Test matrix-vector
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论