Browse Source

!2997 fix validator for InvertPermutation

Merge pull request !2997 from jiangjinsheng/issue_fix4
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
82b4cadad2
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/ops/operations/array_ops.py

+ 3
- 0
mindspore/ops/operations/array_ops.py View File

@@ -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))]


Loading…
Cancel
Save