提交 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):
f1 = inplace_func([avec, bvec], c)
aval = rand(5)
bval = rand(5)
self.assertTrue(numpy.tensordot(aval, bval, axes) == \
f1(aval, bval))
out0 = numpy.tensordot(aval, bval, axes)
out1 = f1(aval, bval)
self.assertTrue(numpy.allclose(out0, out1), (out0, out1))
utt.verify_grad(self.TensorDot(axes), [aval, bval])
# Test matrix-vector
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论