pub fn mseloss(a: &Var, b: &Var) -> Var