Expand description
Loss functions
Structs§
- Binary
Cross Entropy - Binary cross entropy loss.
- Binary
Cross Entropy Builder - Builder for
BinaryCrossEntropy. - Cosine
Similarity Loss - Computes the cosine similarity loss.
- Cosine
Similarity Loss Builder - Builder for
CosineSimilarityLoss. - Cross
Entropy - Cross entropy loss function.
- Cross
Entropy Builder - Builder for
CrossEntropy. - Gaussian
NllLoss - Compute the negative log likelihood loss for a Gaussian distribution.
- Gaussian
NllLoss Builder - Builder for
GaussianNllLoss. - Hinge
Loss - Compute the hinge loss.
- Hinge
Loss Builder - Builder for
HingeLoss. - Huber
Loss - Compute the Huber loss.
- Huber
Loss Builder - Builder for
HuberLoss. - KlDiv
Loss - Compute the Kullback-Leibler divergence loss.
- KlDiv
Loss Builder - Builder for
KlDivLoss. - L1Loss
- Computes the L1 loss
- L1Loss
Builder - Builder for
L1Loss. - LogCosh
Loss - Computes the log cosh loss between inputs and targets.
- LogCosh
Loss Builder - Builder for
LogCoshLoss. - Margin
Ranking Loss - Computes the margin ranking loss.
- Margin
Ranking Loss Builder - Builder for
MarginRankingLoss. - MseLoss
- Computes the mean squared error loss.
- MseLoss
Builder - Builder for
MseLoss. - NllLoss
- Computes the negative log likelihood loss.
- NllLoss
Builder - Builder for
NllLoss. - Smooth
L1Loss - Computes the smooth L1 loss.
- Smooth
L1Loss Builder - Builder for
SmoothL1Loss. - Triplet
Loss - Computes the triplet loss for a set of anchor, positive, and negative samples. Margin is represented with alpha in the math section.
- Triplet
Loss Builder - Builder for
TripletLoss.
Enums§
- Loss
Reduction - Different types of loss reductions
Type Aliases§
- Cross
Entropy Builder Weights - Helper type alias for CrossEntropyBuilder weights.