!2566 Fix some mistakes of ArgMaxWithValue vm ops

Merge pull request !2566 from liuwenhao/arg
5 years ago · 480aecb412
parent a32d674cc5 22eeabdc5f
commit 480aecb412
1 changed files with 1 additions and 1 deletions
--- 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)
    """