ggml_argmax method

Pointer<ggml_tensor> ggml_argmax(Pointer<ggml_context> ctx, Pointer<ggml_tensor> a)

argmax along rows

Implementation

ffi.Pointer<ggml_tensor> ggml_argmax(
  ffi.Pointer<ggml_context> ctx,
  ffi.Pointer<ggml_tensor> a,
) {
  return _ggml_argmax(
    ctx,
    a,
  );
}