diff --git a/mindspore/ops/operations/array_ops.py b/mindspore/ops/operations/array_ops.py index b8f08019d19..fa85e2f3a13 100755 --- a/mindspore/ops/operations/array_ops.py +++ b/mindspore/ops/operations/array_ops.py @@ -1203,7 +1203,7 @@ class Size(PrimitiveWithInfer): else: size = functools.reduce(lambda x, y: x * y, x['shape']) out = {'shape': None, - 'dtype': mstype.int32, + 'dtype': mstype.int64, 'value': size} return out