minor fixes to diag, and some elmentwise ops not returning floats