diff --git a/mindspore/ops/operations/array_ops.py b/mindspore/ops/operations/array_ops.py
index 1bb39d154726d18f6af4ef59a94ee8476e934473..f59a3a37e541e9bf3821957f3c4f9118757c554f 100644
--- a/mindspore/ops/operations/array_ops.py
+++ b/mindspore/ops/operations/array_ops.py
@@ -1123,7 +1123,7 @@ class ArgMaxWithValue(PrimitiveWithInfer):
         - output_x (Tensor) - The maximum value of input tensor, the shape same as index.
 
     Examples:
-        >>> input_x = Tensor(np.random.rand(5))
+        >>> input_x = Tensor(np.random.rand(5), mindspore.float32)
         >>> index, output = P.ArgMaxWithValue()(input_x)
     """