Dynamic number of Dirichlet distributions?

Shape issue is one of our daily struggle :joy: