Joseph Redmon
2014-03-01 1b5e6d8855064e5d990bd73f1f3a0aa00cbdfb4c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
#ifndef SOFTMAX_LAYER_H
#define SOFTMAX_LAYER_H
 
typedef struct {
    int inputs;
    float *delta;
    float *output;
} softmax_layer;
 
softmax_layer *make_softmax_layer(int inputs);
void forward_softmax_layer(const softmax_layer layer, float *input);
void backward_softmax_layer(const softmax_layer layer, float *input, float *delta);
 
#endif