ggml_opt_step_sgd function
- @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_context> ctx, - Pointer<
ggml_tensor> a, - Pointer<
ggml_tensor> grad, - 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,
);