ailoss_mse.h

Go to the documentation of this file.

 
#ifndef MSE_LOSS
#define MSE_LOSS
 
#include "core/aifes_core.h"
 
typedef struct ailoss_mse   ailoss_mse_t; 
struct ailoss_mse {
    ailoss_t base; 
    const aimath_dtype_t *dtype; 
 
    void (*mse_gradients)(const aitensor_t *predicted, const aitensor_t *target, aitensor_t *result);
 
    void (*mse_loss)(const aitensor_t *predicted, const aitensor_t *target, void *result);
 
};
 
extern const aicore_losstype_t *ailoss_mse_type;
 
ailoss_t *ailoss_mse(ailoss_mse_t *loss, ailayer_t *input_layer);
 
void ailoss_mse_calc_delta(ailoss_t *self, const aitensor_t *target_data);
 
void ailoss_mse_calc_loss(ailoss_t *self, const aitensor_t *target_data, void *result);
 
#ifdef AIDEBUG_PRINT_MODULE_SPECS
void ailoss_mse_print_specs(const ailoss_t *self);
#endif // AIDEBUG_PRINT_MODULE_SPECS
 
#endif // MSE_LOSS