p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!1366 InvertPermutation support 1-d tensor input 

Merge pull request !1366 from jiangjinsheng/issue_invert
This commit is contained in:
mindspore-ci-bot 2020-05-22 17:34:53 +08:00 committed by Gitee
parent 479f26ea0d 0fd5d2595c
commit af6a8c1786
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mindspore/ops/operations/array_ops.py
View File

@ -905,9 +905,11 @@ class InvertPermutation(PrimitiveWithInfer):
values can not be negative.
Inputs:
- **input_x** (tuple[int]) - The input tuple is constructed by multiple
- **input_x** (Union(tuple[int], Tensor[int])) - The input tuple is constructed by multiple
integers, i.e., :math:`(y_1, y_2, ..., y_S)` representing the indices.
The values must include 0. There can be no duplicate values or negative values.
If the input is Tensor, it must be 1-d and the dtype is int.
Outputs:
tuple[int]. the lenth is same as input.
@ -926,7 +928,11 @@ class InvertPermutation(PrimitiveWithInfer):
def __infer__(self, x):
x_shp = x['shape']
x_value = x['value']
validator.check_value_type("shape", x_shp, [tuple], self.name)
validator.check_value_type("shape", x_shp, [tuple, list], self.name)
if mstype.issubclass_(x['dtype'], mstype.tensor):
validator.check('x dimension', len(x_shp), '', 1, Rel.EQ, self.name)
validator.check_type_same({'x dtype': x['dtype']}, mstype.int_type, self.name)
x_value = [int(i) for i in x_value.asnumpy()]
z = [x_value[i] for i in range(len(x_value))]
z.sort()