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

flake8, fix test and remove redundant test.

上级 5d0baa3f
......@@ -4746,7 +4746,7 @@ class Reshape(Op):
def __init__(self, ndim, name=None):
self.ndim = ndim
if ndim < 0:
raise ValueError("The output dimensions after reshape must be 0 or greater")
raise ValueError("The output dimensions after reshape must be 0 or greater")
assert name is None, 'name attribute for Reshape has been deprecated'
def __str__(self):
......
......@@ -7632,21 +7632,19 @@ class TestInferShape(utt.InferShapeTester):
# Flatten
atens3 = tensor3()
atens3_val = rand(4, 5, 3)
self._compile_and_check([atens3],
[flatten(atens3, 1)],
[atens3_val], Reshape)
for outdim in (3, 2, 1):
self._compile_and_check([atens3],
[flatten(atens3, outdim)],
[atens3_val], Reshape)
[atens3_val], Reshape,
excluding=['local_useless_reshape'])
amat = matrix()
amat_val = rand(4, 5)
for outdim in (2, 1):
self._compile_and_check([amat],
[flatten(amat, outdim)],
[amat_val], Reshape)
[amat_val], Reshape,
excluding=['local_useless_reshape'])
avec = vector()
avec_val = rand(4)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论