z.moments_diag function

Calculates the first and second moments of the latent data for diagonal models