AIfES 2  2.0.0
ailayer_softmax.h
Go to the documentation of this file.
1 
44 #ifndef SOFTMAX_LAYER
45 #define SOFTMAX_LAYER
46 
47 #include "core/aifes_core.h"
48 
49 typedef struct ailayer_softmax ailayer_softmax_t;
50 
62 
73  void (*softmax)(const aitensor_t *x, aitensor_t *result);
74 
76 };
77 
84 
97 
119 
130 
131 #ifdef AIDEBUG_PRINT_MODULE_SPECS
137 void ailayer_softmax_print_specs(const ailayer_t *self, int (*print)(const char *format, ...));
138 #endif // AIDEBUG_PRINT_MODULE_SPECS
139 
140 #endif // SOFTMAX_LAYER
AIfES 2 core interface.
void ailayer_softmax_calc_result_shape(ailayer_t *self)
Calculate the shape of the result tensor.
void ailayer_softmax_print_specs(const ailayer_t *self, int(*print)(const char *format,...))
Print the layer specification.
const aicore_layertype_t * ailayer_softmax_type
Softmax layer type.
ailayer_t * ailayer_softmax(ailayer_softmax_t *layer, ailayer_t *input_layer)
Initialize and connect the given Softmax layer.
void ailayer_softmax_forward(ailayer_t *self)
Calculate the forward pass for given Softmax layer.
Type indicator of the layer.
Definition: aifes_core.h:81
General Softmax layer struct.
Definition: ailayer_softmax.h:54
void(* softmax)(const aitensor_t *x, aitensor_t *result)
Required math function: Softmax.
Definition: ailayer_softmax.h:73
ailayer_t base
Inherited field members from general ailayer struct.
Definition: ailayer_softmax.h:55
const aimath_dtype_t * dtype
Data type of the input and inference result values.
Definition: ailayer_softmax.h:56
AIfES layer interface.
Definition: aifes_core.h:251
Indicator for the used datatype.
Definition: aifes_math.h:53
A tensor in AIfES.
Definition: aifes_math.h:98