Function std::tensor_cross_entropy_loss

&FloatTensor tensor_cross_entropy_loss(
    const &FloatTensor logits,
    const &FloatTensor labels,
)

Softmax cross-entropy of logits [m, classes] against m class indices, averaged over rows. Gradient to logits only.