fix invertPermutation operator in pynative

5 years ago · 9cfcdce8dc
parent bfa4f8f0dc
commit 9cfcdce8dc
1 changed files with 1 additions and 0 deletions
--- a/mindspore/ops/operations/array_ops.py
+++ b/mindspore/ops/operations/array_ops.py
@ -971,6 +971,7 @@ class InvertPermutation(PrimitiveWithInfer):
    @prim_attr_register
    def __init__(self):
        """init InvertPermutation"""
        self.const_value = True
    def __infer__(self, x):
        x_shp = x['shape']