fix MSE_loss

[luigibonati]

• [ ] Change signature from (diff) to (y,y_ref)
• [ ] Enforce check on the shape of y and y_ref

[luigibonati]

closed as fixed by #37