cross_entropy_with_logits

burn::tensor::loss

Function cross_entropy_with_logits

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

Expand description

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.