提交 619d05f1 authored 作者: Frederic Bastien's avatar Frederic Bastien

flake8, fix test and remove redundant test.

上级 5d0baa3f
...@@ -7632,21 +7632,19 @@ class TestInferShape(utt.InferShapeTester): ...@@ -7632,21 +7632,19 @@ class TestInferShape(utt.InferShapeTester):
# Flatten # Flatten
atens3 = tensor3() atens3 = tensor3()
atens3_val = rand(4, 5, 3) atens3_val = rand(4, 5, 3)
self._compile_and_check([atens3],
[flatten(atens3, 1)],
[atens3_val], Reshape)
for outdim in (3, 2, 1): for outdim in (3, 2, 1):
self._compile_and_check([atens3], self._compile_and_check([atens3],
[flatten(atens3, outdim)], [flatten(atens3, outdim)],
[atens3_val], Reshape) [atens3_val], Reshape,
excluding=['local_useless_reshape'])
amat = matrix() amat = matrix()
amat_val = rand(4, 5) amat_val = rand(4, 5)
for outdim in (2, 1): for outdim in (2, 1):
self._compile_and_check([amat], self._compile_and_check([amat],
[flatten(amat, outdim)], [flatten(amat, outdim)],
[amat_val], Reshape) [amat_val], Reshape,
excluding=['local_useless_reshape'])
avec = vector() avec = vector()
avec_val = rand(4) avec_val = rand(4)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论