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

Neural network layer API. More...

#include <stdbool.h>
#include <stddef.h>
#include "micrograd/neuron.h"
Include dependency graph for layer.h:
This graph shows which files directly or indirectly include this file:

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.
 

Detailed Description

Neural network layer API.