Module loss 

Structs

CrossEntropyLoss

Cross-Entropy Loss with softmax

MAELoss

Mean Absolute Error loss function

MSELoss

Mean Squared Error loss function

Traits

LossFunction

Loss function trait for training neural networks

Functions

softmax

Numerically stable softmax function