Return the mean squared error between pred and target: the average of (pred_i - target_i)^2 over all elements.
pred
target
(pred_i - target_i)^2
solve(...)