Expand description
Loss module
Structs§
- Calculate the binary cross entropy loss from the input logits and the targets.
- Configuration to create a Binary Cross-entropy loss using the init function.
- The record type for the module.
- The record item type for the module.
- Calculate the cross entropy loss from the input logits and the targets.
- Configuration to create a Cross-entropy loss using the init function.
- The record type for the module.
- The record item type for the module.
- Calculate the Huber loss between the inputs and the target.
- Configuration to create a Huber loss.
- Calculate the mean squared error loss from the input logits and the targets.
Enums§
- The reduction type for the loss.