Browse Source

fixed doc for ArgMaxWithValue

tags/v0.5.0-beta
jiangjinsheng 5 years ago
parent
commit
933ef6701a
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      mindspore/ops/operations/array_ops.py

+ 3
- 1
mindspore/ops/operations/array_ops.py View File

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


Loading…
Cancel
Save