提交 82b4cada 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!2997 fix validator for InvertPermutation

Merge pull request !2997 from jiangjinsheng/issue_fix4
......@@ -1016,6 +1016,9 @@ class InvertPermutation(PrimitiveWithInfer):
validator.check_value_type("shape", x_shp, [tuple, list], self.name)
if mstype.issubclass_(x['dtype'], mstype.tensor):
raise ValueError(f'For \'{self.name}\' the input value must be non-Tensor.')
for shp in x_shp:
if shp != []:
raise ValueError(f'For \'{self.name}\' the rank of input must be 1.')
for i, value in enumerate(x_value):
validator.check_value_type("input[%d]" % i, value, [int], self.name)
z = [x_value[i] for i in range(len(x_value))]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册