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

Test a few other shape/axis combinations

上级 cd2ffc37
......@@ -164,8 +164,11 @@ class test_CAReduce(unittest.TestCase):
((5, 6), ()),
((2, 3, 4, 5), (0, 1, 3)),
((2, 3, 4, 5), (-2, -3)),
((5, 0), None),
((5, 0), (0, )),
((5, 0), (1, )),
((5, 0), ()),
((), None),
((), ())]:
x = TensorType('float64', [(entry == 1) for entry in xsh])('x')
e = CAReduce(scalar_op, axis = tosum)(x)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论