The dimension 2 (dim2) usually generates longer and informative responses.
following the techniques described above, ELBO then has the form: where and the KL-divergence term can be derived as * : where α, β are parameters of Dirichlet distribution q φ (z|x) and p(z) respectively, K is the dimension of z, and ψ is the Digamma function.