topk function

List<Tensor> topk(
  1. Tensor a,
  2. int k, {
  3. int dim = -1,
  4. bool largest = true,
  5. bool sorted = true,
})

Implementation

List<Tensor> topk(Tensor a, int k,
    {int dim = -1, bool largest = true, bool sorted = true}) {
  return a.topk(k, dim: dim, largest: largest, sorted: sorted);
}