ggml_opt_step_sgd function

  1. @Native<Pointer<ggml_tensor> Function(Pointer<ggml_context>, Pointer<ggml_tensor>, Pointer<ggml_tensor>, Pointer<ggml_tensor>)>(ffi.Pointer<ggml_context>, ffi.Pointer<ggml_tensor>, ffi.Pointer<ggml_tensor>, ffi.Pointer<ggml_tensor>)>()
Pointer<ggml_tensor> ggml_opt_step_sgd(
  1. Pointer<ggml_context> ctx,
  2. Pointer<ggml_tensor> a,
  3. Pointer<ggml_tensor> grad,
  4. Pointer<ggml_tensor> sgd_params,
)

stochastic gradient descent step (with weight decay)

Implementation

@ffi.Native<
  ffi.Pointer<ggml_tensor> Function(
    ffi.Pointer<ggml_context>,
    ffi.Pointer<ggml_tensor>,
    ffi.Pointer<ggml_tensor>,
    ffi.Pointer<ggml_tensor>,
  )
>()
external ffi.Pointer<ggml_tensor> ggml_opt_step_sgd(
  ffi.Pointer<ggml_context> ctx,
  ffi.Pointer<ggml_tensor> a,
  ffi.Pointer<ggml_tensor> grad,
  ffi.Pointer<ggml_tensor> sgd_params,
);