Merge #1795

bors[bot] · staticfloat · web-flow · commit 48596ef45f73 · 2021-12-13T18:02:43.000Z
1795: Simplify mse() to use `abs2()` r=ToucheSir a=staticfloat

This simplifies the implementation of `mse()` a bit, as requested by `@DhairyaLGandhi` 

Co-authored-by: Elliot Saba &lt;staticfloat@gmail.com&gt;
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
 
 """
