cross_entropy_stable

Function cross_entropy_stable 

Source
pub fn cross_entropy_stable<T: Float>(
    predictions: &[T],
    targets: &[T],
) -> CoreResult<T>
Expand description

Cross entropy loss with numerical stability