ggml_top_k method

Pointer<ggml_tensor> ggml_top_k(Pointer<ggml_context> ctx, Pointer<ggml_tensor> a, int k)

top k elements per row

Implementation

ffi.Pointer<ggml_tensor> ggml_top_k(
  ffi.Pointer<ggml_context> ctx,
  ffi.Pointer<ggml_tensor> a,
  int k,
) {
  return _ggml_top_k(
    ctx,
    a,
    k,
  );
}