#include "softmax_layer.h"#include "blas.h"#include "cuda.h"#include <float.h>#include <math.h>#include <stdlib.h>#include <stdio.h>#include <assert.h>
Include dependency graph for softmax_layer.c:
Go to the source code of this file.
Functions | |
| void | backward_softmax_layer (const softmax_layer l, network_state state) |
| void | forward_softmax_layer (const softmax_layer l, network_state state) |
| softmax_layer | make_softmax_layer (int batch, int inputs, int groups) |
| void | softmax_tree (float *input, int batch, int inputs, float temp, tree *hierarchy, float *output) |
Function Documentation
| void backward_softmax_layer | ( | const softmax_layer | l, |
| network_state | state | ||
| ) |
Definition at line 63 of file softmax_layer.c.
| void forward_softmax_layer | ( | const softmax_layer | l, |
| network_state | state | ||
| ) |
Definition at line 49 of file softmax_layer.c.
| softmax_layer make_softmax_layer | ( | int | batch, |
| int | inputs, | ||
| int | groups | ||
| ) |
Definition at line 10 of file softmax_layer.c.
| void softmax_tree | ( | float * | input, |
| int | batch, | ||
| int | inputs, | ||
| float | temp, | ||
| tree * | hierarchy, | ||
| float * | output | ||
| ) |
Definition at line 35 of file softmax_layer.c.