TopKResult typedef

TopKResult = (TensorBuffer, TensorBuffer)

The result of a TopK operation: (values, indices).

• values: A TensorBuffer containing the k largest or smallest values.
• indices: A TensorBuffer (DType.int64) containing the original indices of the selected values along the specified axis.

Implementation

typedef TopKResult = (TensorBuffer values, TensorBuffer indices);