Returns the maximum value and its index position in the specified dimension of a tensor. The maximum value is saved to out, and its index is saved to indices. If [object Object] is set to [object Object], the corresponding axis is not retained. If [object Object] is set to [object Object], the dimension value [object Object] of the specified axis is retained.
Each operator has calls. First, [object Object] is called to obtain the workspace size required for computation and the executor that contains the operator computation process. Then, [object Object] is called to perform computation.
Parameter description:
[object Object]- [object Object]Atlas inference products[object Object] and [object Object]Atlas training products[object Object]:
- The self and out parameters do not support the BFLOAT16 and INT32 data types.
- The indices parameter does not support the BFLOAT16 data type.
- [object Object]Atlas A2 training products/Atlas A2 inference products[object Object] and [object Object]Atlas A3 training products/Atlas A3 inference products[object Object]:
- The self and out parameters do not support the INT32 data type.
- [object Object]Atlas inference products[object Object] and [object Object]Atlas training products[object Object]:
Returns:
[object Object]: status code. For details, see .The first-phase API performs input parameter validation. The following errors may be returned:
[object Object]
- Deterministic computation:
[object Object]defaults to a deterministic implementation.
The following example is for reference only. For details, see .