let
square_loss y y' =
let
open
MX
in
average_rows ((y' -@ y) **@ 2.)