Module dfdx::losses

Standard loss functions such as mse, mae, cross entropy, and more.

Functions

cross_entropy_with_logits_loss

Cross entropy loss. This will call log_softmax(logits), so make sure logits is not the output from softmax() or log_softmax() already.

mae_loss

Mean absolute error. This is the same as doing (pred - &targ).abs().mean()

mse_loss

Mean Squared Error. This is the same as doing (pred - &targ).square().mean()

rmse_loss

Root Mean square error. This is the same as doing (pred - &targ).square().mean().sqrt()