diff --git a/mindspore/ops/operations/array_ops.py b/mindspore/ops/operations/array_ops.py index 16f5121e65..e23a242820 100644 --- a/mindspore/ops/operations/array_ops.py +++ b/mindspore/ops/operations/array_ops.py @@ -1105,9 +1105,11 @@ class ArgMaxWithValue(PrimitiveWithInfer): :math:`(x_1, x_2, ..., x_N)`. Outputs: - Tensor, corresponding index and maximum value of input tensor. If `keep_dims` is true, the output tensors shape + tuple(Tensor), tuple of 2 tensors, corresponding index and maximum value of input tensor. + - index (Tensor) - The index for maximum value of input tensor. If `keep_dims` is true, the output tensors shape is :math:`(x_1, x_2, ..., x_{axis-1}, 1, x_{axis+1}, ..., x_N)`. Else, the shape is :math:`(x_1, x_2, ..., x_{axis-1}, x_{axis+1}, ..., x_N)`. + - output_x (Tensor) - The maximum value of input tensor, the shape same as index. Examples: >>> input_x = Tensor(np.random.rand(5))