Function cross_entropy_with_logits 

pub fn cross_entropy_with_logits<const D: usize>(
    logits: Tensor<D>,
    target_probs: Tensor<D>,
) -> Tensor<1>

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

Arguments

• logits - The logits.
• target_probs - The target probabilities.

Returns

The log softmax cross entropy.