Skip to documentation

Frameworks

Loss Functions

Loss Functions Built in loss functions Topics Loss Functions binaryCrossEntropy(logits:targets:weights:withLogits:reduction:) cosineSimilarityLoss(x1:x2:axis:eps:reduction:) crossEntropy(logits:targets:weights:axis:labelSmoothing:reduction:

Loss Functions

Built-in loss functions

Topics

Loss Functions

  • binaryCrossEntropy(logits:targets:weights:withLogits:reduction:)
  • cosineSimilarityLoss(x1:x2:axis:eps:reduction:)
  • crossEntropy(logits:targets:weights:axis:labelSmoothing:reduction:)
  • hingeLoss(inputs:targets:reduction:)
  • huberLoss(inputs:targets:delta:reduction:)
  • klDivLoss(inputs:targets:axis:reduction:)
  • l1Loss(predictions:targets:reduction:)
  • logCoshLoss(inputs:targets:reduction:)
  • mseLoss(predictions:targets:reduction:)
  • nllLoss(inputs:targets:axis:reduction:)
  • smoothL1Loss(predictions:targets:beta:reduction:)
  • tripletLoss(anchors:positives:negatives:axis:p:margin:eps:reduction:)