Simplify mse() to use abs2()

This simplifies the implementation of `mse()` a bit.
FluxML · Dec 1, 2021 · f2998cf · f2998cf
1 parent 8fa4317
commit f2998cf
Showing 1 changed file with 1 addition and 2 deletions.
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
 
 """