Probability distributions - torch.distributions

Distribution

Bernoulli

Categorical

Normal