burn::tensor

Module loss

Expand description

The loss module.

Functionsยง

cross_entropy_with_logits
Computes the log softmax cross entropy between logits and target probabilities.