pub fn mse_loss<T: Reduce<AllAxes>>(pred: T, targ: T::NoTape) -> T::Reduced
Expand description

Mean Squared Error. This computes (pred - targ).square().mean().

See mean(), square(), and sub().