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?