micrograd 0.1.0
Small C implementation of micrograd
Loading...
Searching...
No Matches
mlp.h File Reference

Multi-layer perceptron API. More...

#include <stddef.h>
#include "micrograd/layer.h"
Include dependency graph for mlp.h:

Go to the source code of this file.

Data Structures

struct  mg_mlp
 Multi-layer perceptron model. More...
 

Functions

bool mg_mlp_init (mg_graph *g, mg_mlp *m, size_t n_in, const size_t *sizes, size_t n_sizes)
 Initialize an MLP model.
 
bool mg_mlp_call (mg_graph *g, mg_mlp *m, mg_value **x, mg_value **out)
 Evaluate an MLP model.
 
void mg_mlp_free (mg_mlp *m)
 Free memory owned by an MLP model.
 
size_t mg_mlp_param_count (const mg_mlp *m)
 Get the number of trainable parameters in an MLP model.
 
void mg_mlp_params (const mg_mlp *m, mg_value **out)
 Copy an MLP model's parameters into a flat array.
 

Detailed Description

Multi-layer perceptron API.