just a simple dim check
Latent Dirich. Alloc. collapsed t'other way around

There are still some terms from the KL divergence missing, so the bound
does weird things, but the factorisation works well