Structs§
- Cross
Entropy Loss - Cross-Entropy Loss with softmax
- MAELoss
- Mean Absolute Error loss function
- MSELoss
- Mean Squared Error loss function
Traits§
- Loss
Function - Loss function trait for training neural networks
Functions§
- softmax
- Numerically stable softmax function