pub fn cross_entropy_stable<T: Float>(
predictions: &[T],
targets: &[T],
) -> CoreResult<T>
Expand description
Cross entropy loss with numerical stability
pub fn cross_entropy_stable<T: Float>(
predictions: &[T],
targets: &[T],
) -> CoreResult<T>
Cross entropy loss with numerical stability