[object Object]

[object Object][object Object]undefined[object Object]

Description: Returns the maximum value of a specified dimension of a tensor and its index. 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.

[object Object]

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.

• [object Object]
• [object Object]

[object Object]

• Parameter description:

  • [object Object] (aclTensor*, computation input): aclTensor on the device. are supported. The can be ND. The [1, 8] dimensions are supported. [object Object]Atlas inference series products[object Object] and [object Object]Atlas training series products[object Object]: The data type can be FLOAT, FLOAT16, INT64, or BOOL. [object Object]Atlas A2 training products/Atlas A2 inference products[object Object] and [object Object]Atlas A3 training series products/Atlas A3 inference series products[object Object]: The data type can be FLOAT, FLOAT16, INT64, BOOL, or BFLOAT16.

  • [object Object] (int64_t, compute input): specified dimension. The data type is INT64. The value range is [–self.dim(), self.dim()).

  • [object Object] (bool, compute input): whether to retain the dimension of the reduced axis. The data type is BOOL.

  • [object Object] (aclTensor*, compute output): aclTensor on the device. The data type is the same as that of [object Object]. are supported. The can be ND. If [object Object] is [object Object], the number of output dimensions is that of [object Object] minus 1. If [object Object] is [object Object], the number of output dimensions is equal to that of [object Object]. [object Object]Atlas inference series products[object Object] and [object Object]Atlas training series products[object Object]: The data type can be FLOAT, FLOAT16, INT64, or BOOL. [object Object]Atlas A2 training products/Atlas A2 inference products[object Object] and [object Object]Atlas A3 training series products/Atlas A3 inference series products[object Object]: The data type can be FLOAT, FLOAT16, INT64, BOOL, or BFLOAT16.

  • [object Object] (aclTensor*, compute output): aclTensor on the device. are supported. The can be ND. [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 data type can be BOOL, FLOAT32, FLOAT16, INT8, INT16, UINT16, UINT8, INT32, INT64, UINT32, UINT64, DOUBLE, COMPLEX64, COMPLEX128, or BFLOAT16

    • [object Object]Atlas inference products[object Object] and [object Object]Atlas training products[object Object]: The data type can be BOOL, FLOAT32, FLOAT16, INT8, INT16, UINT16, UINT8, INT32, INT64, UINT32, UINT64, DOUBLE, COMPLEX64, or COMPLEX128.

  • [object Object] (uint64_t*, output): size of the workspace to be allocated on the device.

  • [object Object] (aclOpExecutor**, output): operator executor, containing the operator computation process.

• Returns:

  [object Object]: status code. For details, see .

[object Object]

• Parameter description:

  • [object Object] (void*, input): address of the workspace to be allocated on the device.

  • [object Object] (uint64_t, input): size of the workspace to be allocated on the device, which is obtained by calling [object Object].

  • [object Object] (aclOpExecutor*, input): operator executor, containing the operator computation process.

  • [object Object] (aclrtStream, input): stream for executing the task.

• Returns:

  [object Object]: status code. For details, see .

[object Object]

• Deterministic computation:
  • [object Object] defaults to a deterministic implementation.

[object Object]

The following example is for reference only. For details, see .