提交 741ae918 authored 作者: Frederic Bastien's avatar Frederic Bastien

Give a better test error and be more resilent to eps changes in implementations.

上级 dbe33b01
...@@ -6583,8 +6583,8 @@ class test_tensordot(unittest.TestCase): ...@@ -6583,8 +6583,8 @@ class test_tensordot(unittest.TestCase):
f3 = inplace_func([amat, bmat], c) f3 = inplace_func([amat, bmat], c)
aval = rand(4, 7) aval = rand(4, 7)
bval = rand(7, 9) bval = rand(7, 9)
self.assertTrue(np.allclose(np.tensordot(aval, bval, axes), utt.assert_allclose(np.tensordot(aval, bval, axes),
f3(aval, bval))) f3(aval, bval))
utt.verify_grad(self.TensorDot(axes), [aval, bval]) utt.verify_grad(self.TensorDot(axes), [aval, bval])
def test_scalar_axes(self): def test_scalar_axes(self):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论