Enums§
- Reduction
- Reduction mode for loss functions.
Functions§
- bce_
loss - Binary Cross-Entropy loss for probabilities in [0, 1].
- bce_
with_ logits_ loss - Binary Cross-Entropy with Logits (numerically stable).
- cross_
entropy_ loss - Cross-entropy loss with log-softmax for numerical stability.
- l1_loss
- L1 Loss (Mean Absolute Error): mean(|prediction - target|)
- l1_
loss_ with_ reduction - L1 Loss with configurable reduction.
- mse_
loss - Mean Squared Error loss: mean((prediction - target)²)
- mse_
loss_ with_ reduction - MSE Loss with configurable reduction.
- nll_
loss - Negative Log-Likelihood Loss with integer class indices.
- smooth_
l1_ loss - Smooth L1 Loss (Huber Loss):