Module burn::tensor::loss

The loss module.

Functions

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