|
micrograd 0.1.0
Small C implementation of micrograd
|
Neural network layer API. More...


Go to the source code of this file.
Data Structures | |
| struct | mg_layer |
| Layer of neurons in a neural network. More... | |
Functions | |
| bool | mg_layer_init (mg_graph *g, mg_layer *l, size_t n_in, size_t n_out, bool non_linear) |
| Initialize a layer. | |
| bool | mg_layer_call (mg_graph *g, mg_layer *l, mg_value **x, mg_value **out) |
| Evaluate a layer with the given input values. | |
| void | mg_layer_free (mg_layer *l) |
| Free memory owned by a layer. | |
| size_t | mg_layer_param_count (const mg_layer *l) |
| Get the number of trainable parameters in a layer. | |
| void | mg_layer_params (const mg_layer *l, mg_value **out) |
| Copy a layer's parameters into a flat array. | |
Neural network layer API.