ailoss_mse Struct Reference

General Mean Squared Error (MSE) loss struct. More...

#include <ailoss_mse.h>

Data Fields
ailoss_t	base
	Inherited field members from general ailoss struct.
const aimath_dtype_t *	dtype
	Main data type of the loss.
Math fuctions
Required data type specific math functions
void(*	mse_gradients )(const aitensor_t *predicted, const aitensor_t *target, aitensor_t *result)
	Required math function: Gradient calculation of MSE loss function. More...
void(*	mse_loss )(const aitensor_t *predicted, const aitensor_t *target, void *result)
	Required math function: Calculation of the MSE loss function. More...

Detailed Description

General Mean Squared Error (MSE) loss struct.

Field Documentation

mse_gradients

void(* mse_gradients) (const aitensor_t *predicted, const aitensor_t *target, aitensor_t *result)

Required math function: Gradient calculation of MSE loss function.

Requires a math function that calculates the gradient for each output with respect to the MSE loss:

mse_loss

void(* mse_loss) (const aitensor_t *predicted, const aitensor_t *target, void *result)

Required math function: Calculation of the MSE loss function.

Requires a math function that calculates loss of the prediction compared to the target:

---

The documentation for this struct was generated from the following file:

• ailoss_mse.h