afp_topk

Returns the top k values along a given dimension of the input array. The elements are extracted along dimension 1 (columns).

Arguments

  1. Input array, values
  2. Integer number k, number of elements to return
  3. Integer, 0 - ascending, 1 - descending

Output

Resulting value or array

Examples

afp_topk