Distributions

class torchelie.distributions.GaussianMixture(weights, locs, scales)

Mixture of gaussian distributions. Each tensor contains an additional dimension with number of distributions elements.

Parameters:

• weights (tensor) – un-normalized weights of distributions
• loc (tensor) – mean of the distributions
• scale (tensor) – scale of the distributions
• dim (int) – dimension reprenseting the various distributions, that will weighted and averaged on.

log_prob(x)

mean

class torchelie.distributions.Logistic(loc, scale)

Logistic distribution

Parameters:

• loc (tensor) – mean of the distribution
• scale (tensor) – scale of the distribution

class torchelie.distributions.LogisticMixture(weights, locs, scales, dim)

Mixture of Logistic distributions. Each tensor contains an additional dimension with number of distributions elements.

Parameters:

• weights (tensor) – un-normalized weights of distributions
• loc (tensor) – mean of the distributions
• scale (tensor) – scale of the distributions
• dim (int) – dimension reprenseting the various distributions, that will weighted and averaged on.

log_prob(x)

mean