!2566 Fix some mistakes of ArgMaxWithValue vm ops

Merge pull request !2566 from liuwenhao/arg

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)
     """