Simplify mse() to use abs2()

staticfloat · web-flow · commit f2998cf5e6c8 · 2021-11-30T17:04:34.000-08:00
This simplifies the implementation of `mse()` a bit.
diff --git a/src/losses/functions.jl b/src/losses/functions.jl
@@ -44,8 +44,7 @@ julia> Flux.mse(y_model, y_true)
 """
 function mse(ŷ, y; agg = mean)
   _check_sizes(ŷ, y)
-  error = ŷ .- y
-  real(agg(error .* conj(error)))
+  agg(abs2.(ŷ .- y)))
 end
 
 """
