ailoss_crossentropy.h

Go to the documentation of this file.

 
#ifndef CROSSENTROPY_LOSS
#define CROSSENTROPY_LOSS
 
#include "core/aifes_core.h"
#include "basic/base/ailayer/ailayer_softmax.h"
#include "basic/base/ailayer/ailayer_sigmoid.h"
 
typedef struct ailoss_crossentropy  ailoss_crossentropy_t; 
struct ailoss_crossentropy {
    ailoss_t base; 
    const aimath_dtype_t *dtype; 
 
    void (*tensor_sub)(const aitensor_t *a, const aitensor_t *b, aitensor_t *result);
 
    void (*crossentropy)(const aitensor_t *predicted_data, const aitensor_t *target_data, void *result);
 
    void (*scale_by_batch_size)(const aitensor_t *predicted_data, aitensor_t *result);
 
};
 
extern const aicore_losstype_t *ailoss_crossentropy_type;
 
ailoss_t *ailoss_crossentropy(ailoss_crossentropy_t *loss, ailayer_t *input_layer);
 
void ailoss_crossentropy_calc_delta(ailoss_t *self, const aitensor_t *target_data);
 
void ailoss_crossentropy_calc_loss(ailoss_t *self, const aitensor_t *target_data, void *result);
 
void ailoss_crossentropy_dummy_backward(ailayer_t *self);
 
#ifdef AIDEBUG_PRINT_MODULE_SPECS
void ailoss_crossentropy_print_specs(const ailoss_t *self);
#endif // AIDEBUG_PRINT_MODULE_SPECS
 
#endif // CROSSENTROPY_LOSS