mabor::tensor

Module loss

Expand description

The loss module.

Functions§

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