Module losses

Loss functions

Structs

• BinaryCrossEntropy
  Binary cross entropy loss.
• BinaryCrossEntropyBuilder
  Builder for BinaryCrossEntropy.
• CosineSimilarityLoss
  Computes the cosine similarity loss.
• CosineSimilarityLossBuilder
  Builder for CosineSimilarityLoss.
• CrossEntropy
  Cross entropy loss function.
• CrossEntropyBuilder
  Builder for CrossEntropy.
• GaussianNllLoss
  Compute the negative log likelihood loss for a Gaussian distribution.
• GaussianNllLossBuilder
  Builder for GaussianNllLoss.
• HingeLoss
  Compute the hinge loss.
• HingeLossBuilder
  Builder for HingeLoss.
• HuberLoss
  Compute the Huber loss.
• HuberLossBuilder
  Builder for HuberLoss.
• KlDivLoss
  Compute the Kullback-Leibler divergence loss.
• KlDivLossBuilder
  Builder for KlDivLoss.
• L1Loss
  Computes the L1 loss
• L1LossBuilder
  Builder for L1Loss.
• LogCoshLoss
  Computes the log cosh loss between inputs and targets.
• LogCoshLossBuilder
  Builder for LogCoshLoss.
• MarginRankingLoss
  Computes the margin ranking loss.
• MarginRankingLossBuilder
  Builder for MarginRankingLoss.
• MseLoss
  Computes the mean squared error loss.
• MseLossBuilder
  Builder for MseLoss.
• NllLoss
  Computes the negative log likelihood loss.
• NllLossBuilder
  Builder for NllLoss.
• SmoothL1Loss
  Computes the smooth L1 loss.
• SmoothL1LossBuilder
  Builder for SmoothL1Loss.
• TripletLoss
  Computes the triplet loss for a set of anchor, positive, and negative samples. Margin is represented with alpha in the math section.
• TripletLossBuilder
  Builder for TripletLoss.

Enums

• LossReduction
  Different types of loss reductions

Type Aliases

• CrossEntropyBuilderWeights
  Helper type alias for CrossEntropyBuilder weights.