mirror of https://github.com/AlexeyAB/darknet.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
15 lines
371 B
15 lines
371 B
#ifndef SOFTMAX_LAYER_H |
|
#define SOFTMAX_LAYER_H |
|
|
|
typedef struct { |
|
int inputs; |
|
int batch; |
|
float *delta; |
|
float *output; |
|
} softmax_layer; |
|
|
|
softmax_layer *make_softmax_layer(int batch, 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
|
|
|