Skip to main content
Module cross_entropy
burn_
mamba
0.20.0
Module cross_
entropy
Module Items
Structs
In burn_
mamba::
utils::
loss
burn_mamba
::
utils
::
loss
Module
cross_
entropy
Copy item path
Source
Structs
ยง
Cross
Entropy
Loss
Calculate the cross-entropy loss from the output logits and the targets.
Cross
Entropy
Loss
Config
Configuration to create a
Cross-entropy loss
using the
init function
.