提交 cfb66e1d authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Compare size, not just value, in test

上级 555af254
......@@ -323,7 +323,9 @@ class test_CAReduce(unittest_tools.InferShapeTester):
zv),
(f(xv), zv))
else:
self.assertTrue(numpy.allclose(f(xv), zv), (f(xv), zv))
f_xv = f(xv)
self.assertTrue((f_xv.shape == zv.shape), (f_xv, zv))
self.assertTrue(numpy.allclose(f_xv, zv), (f_xv, zv))
#test CAReduce.infer_shape
#the Shape op don't implement c_code!
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论