Dirichlet distribution is giving nan and inf

dirichlet distribution is giving nan and inf

Please give details about your problem, what you tried and ideally a code snippet reproducing the issue. Also consider posting any distribution specific questions on the PyTorch slack or forum.