Easier initialization for BayesDiagNormal