[object Object]

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

Description: Returns the top k values and indexes of the input tensor along the specified dimension.

[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]

• Parameters

  • [object Object] (aclTensor*, compute input): aclTensor on the device. The shape supports 1 to 8 dimensions. are supported. The can be ND.
    • [object Object]Atlas inference products[object Object] and [object Object]Atlas training products[object Object]: The data type can be INT8, UINT8, INT16, INT32, INT64, FLOAT16, FLOAT32, or DOUBLE.
    • [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 INT8, UINT8, INT16, INT32, INT64, FLOAT16, FLOAT32, DOUBLE, or BFLOAT16.
  • [object Object] (int64_t, compute input): integer on the host. It indicates the number of top k values along the specified dimension. The value range is [0, self.size(dim)].
  • [object Object] (int64_t, compute input): integer on the host. It indicates the computation dimension. The value range is [-self.dim(), self.dim()].
  • [object Object] (bool, compute input): Boolean type on the host. [object Object] indicates that the output result in the computation dimension is sorted in descending order. [object Object] indicates that the output result in the computation dimension is sorted in ascending order.
  • [object Object] (bool, compute input): Boolean type on the host. [object Object] indicates that the output result is sorted (if [object Object] is [object Object], the result is sorted in descending order; otherwise, the result is sorted in ascending order). [object Object] indicates that the output result is not sorted and is output based on the input data sequence. Note: Currently, this parameter can only be set to [object Object].
  • [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. The sorting axis of the shape is the same as that of [object Object], and non-sorting axes are the same as those of [object Object].
    • [object Object]Atlas inference products[object Object] and [object Object]Atlas training products[object Object]: The data type can be INT8, UINT8, INT16, INT32, INT64, FLOAT16, FLOAT32, or DOUBLE.
    • [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 INT8, UINT8, INT16, INT32, INT64, FLOAT16, FLOAT32, DOUBLE, or BFLOAT16.
  • [object Object] (aclTensor*, compute output): aclTensor on the device. The data type can be INT64. are supported. The can be ND. The sorting axis of the shape is the same as that of [object Object], and non-sorting axes are the same as those of [object Object].
  • [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]

• Parameters

  • [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 the first-phase API [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 .