Convolutional layers working w/ matrices

This commit is contained in:
Joseph Redmon
2014-01-28 16:28:42 -08:00
parent b2b7137b6f
commit f7a17f82eb
26 changed files with 459 additions and 363 deletions

View File

@ -6,14 +6,14 @@
typedef struct {
int h,w,c;
int stride;
double *delta;
double *output;
float *delta;
float *output;
} maxpool_layer;
image get_maxpool_image(maxpool_layer layer);
maxpool_layer *make_maxpool_layer(int h, int w, int c, int stride);
void forward_maxpool_layer(const maxpool_layer layer, double *in);
void backward_maxpool_layer(const maxpool_layer layer, double *in, double *delta);
void forward_maxpool_layer(const maxpool_layer layer, float *in);
void backward_maxpool_layer(const maxpool_layer layer, float *in, float *delta);
#endif