Skip to main content

Module cross_entropy

Module cross_entropy 

Source

Structsยง

CrossEntropyLoss
Calculate the cross-entropy loss from the output logits and the targets.
CrossEntropyLossConfig
Configuration to create a Cross-entropy loss using the init function.