How generate batch of mixture distributions of Gaussian

I wanna generate a batch of mixture distribution of Gaussian from MLP’s output which represents the log of sigma, means, and logits. Are there any APIs in the distribution module work for this? Or just calculate from the formula?

This looks like a distribution instead of distributed question. cc @ptrblck for help :slight_smile:

I have solved it, thanks for your reply.