Skip to content

Latest commit

 

History

History
387 lines (298 loc) · 7.86 KB

distributions.rst

File metadata and controls

387 lines (298 loc) · 7.86 KB
Raw

Probability distributions - torch.distributions

.. automodule:: torch.distributions

.. currentmodule:: torch.distributions

Distribution

.. currentmodule:: torch.distributions.distribution

.. autoclass:: Distribution
    :members:
    :show-inheritance:

ExponentialFamily

.. currentmodule:: torch.distributions.exp_family

.. autoclass:: ExponentialFamily
    :members:
    :show-inheritance:

Bernoulli

.. currentmodule:: torch.distributions.bernoulli

.. autoclass:: Bernoulli
    :members:
    :undoc-members:
    :show-inheritance:

Beta

.. currentmodule:: torch.distributions.beta

.. autoclass:: Beta
    :members:
    :undoc-members:
    :show-inheritance:

Binomial

.. currentmodule:: torch.distributions.binomial

.. autoclass:: Binomial
    :members:
    :undoc-members:
    :show-inheritance:

Categorical

.. currentmodule:: torch.distributions.categorical

.. autoclass:: Categorical
    :members:
    :undoc-members:
    :show-inheritance:

Cauchy

.. currentmodule:: torch.distributions.cauchy

.. autoclass:: Cauchy
    :members:
    :undoc-members:
    :show-inheritance:

Chi2

.. currentmodule:: torch.distributions.chi2

.. autoclass:: Chi2
    :members:
    :undoc-members:
    :show-inheritance:

ContinuousBernoulli

.. currentmodule:: torch.distributions.continuous_bernoulli

.. autoclass:: ContinuousBernoulli
    :members:
    :undoc-members:
    :show-inheritance:

Dirichlet

.. currentmodule:: torch.distributions.dirichlet

.. autoclass:: Dirichlet
    :members:
    :undoc-members:
    :show-inheritance:

Exponential

.. currentmodule:: torch.distributions.exponential

.. autoclass:: Exponential
    :members:
    :undoc-members:
    :show-inheritance:

FisherSnedecor

.. currentmodule:: torch.distributions.fishersnedecor

.. autoclass:: FisherSnedecor
    :members:
    :undoc-members:
    :show-inheritance:

Gamma

.. currentmodule:: torch.distributions.gamma

.. autoclass:: Gamma
    :members:
    :undoc-members:
    :show-inheritance:

Geometric

.. currentmodule:: torch.distributions.geometric

.. autoclass:: Geometric
    :members:
    :undoc-members:
    :show-inheritance:

Gumbel

.. currentmodule:: torch.distributions.gumbel

.. autoclass:: Gumbel
    :members:
    :undoc-members:
    :show-inheritance:

HalfCauchy

.. currentmodule:: torch.distributions.half_cauchy

.. autoclass:: HalfCauchy
    :members:
    :undoc-members:
    :show-inheritance:

HalfNormal

.. currentmodule:: torch.distributions.half_normal

.. autoclass:: HalfNormal
    :members:
    :undoc-members:
    :show-inheritance:

Independent

.. currentmodule:: torch.distributions.independent

.. autoclass:: Independent
    :members:
    :undoc-members:
    :show-inheritance:

Kumaraswamy

.. currentmodule:: torch.distributions.kumaraswamy

.. autoclass:: Kumaraswamy
    :members:
    :undoc-members:
    :show-inheritance:

LKJCholesky

.. currentmodule:: torch.distributions.lkj_cholesky

.. autoclass:: LKJCholesky
    :members:
    :undoc-members:
    :show-inheritance:

Laplace

.. currentmodule:: torch.distributions.laplace

.. autoclass:: Laplace
    :members:
    :undoc-members:
    :show-inheritance:

LogNormal

.. currentmodule:: torch.distributions.log_normal

.. autoclass:: LogNormal
    :members:
    :undoc-members:
    :show-inheritance:

LowRankMultivariateNormal

.. currentmodule:: torch.distributions.lowrank_multivariate_normal

.. autoclass:: LowRankMultivariateNormal
    :members:
    :undoc-members:
    :show-inheritance:

MixtureSameFamily

.. currentmodule:: torch.distributions.mixture_same_family

.. autoclass:: MixtureSameFamily
    :members:
    :undoc-members:
    :show-inheritance:

Multinomial

.. currentmodule:: torch.distributions.multinomial

.. autoclass:: Multinomial
    :members:
    :undoc-members:
    :show-inheritance:

MultivariateNormal

.. currentmodule:: torch.distributions.multivariate_normal

.. autoclass:: MultivariateNormal
    :members:
    :undoc-members:
    :show-inheritance:

NegativeBinomial

.. currentmodule:: torch.distributions.negative_binomial

.. autoclass:: NegativeBinomial
    :members:
    :undoc-members:
    :show-inheritance:

Normal

.. currentmodule:: torch.distributions.normal

.. autoclass:: Normal
    :members:
    :undoc-members:
    :show-inheritance:

OneHotCategorical

.. currentmodule:: torch.distributions.one_hot_categorical

.. autoclass:: OneHotCategorical
    :members:
    :undoc-members:
    :show-inheritance:

Pareto

.. currentmodule:: torch.distributions.pareto

.. autoclass:: Pareto
    :members:
    :undoc-members:
    :show-inheritance:

Poisson

.. currentmodule:: torch.distributions.poisson

.. autoclass:: Poisson
    :members:
    :undoc-members:
    :show-inheritance:

RelaxedBernoulli

.. currentmodule:: torch.distributions.relaxed_bernoulli

.. autoclass:: RelaxedBernoulli
    :members:
    :undoc-members:
    :show-inheritance:

LogitRelaxedBernoulli

.. currentmodule:: torch.distributions.relaxed_bernoulli

.. autoclass:: LogitRelaxedBernoulli
    :members:
    :undoc-members:
    :show-inheritance:

RelaxedOneHotCategorical

.. currentmodule:: torch.distributions.relaxed_categorical

.. autoclass:: RelaxedOneHotCategorical
    :members:
    :undoc-members:
    :show-inheritance:

StudentT

.. currentmodule:: torch.distributions.studentT

.. autoclass:: StudentT
    :members:
    :undoc-members:
    :show-inheritance:

TransformedDistribution

.. currentmodule:: torch.distributions.transformed_distribution

.. autoclass:: TransformedDistribution
    :members:
    :undoc-members:
    :show-inheritance:

Uniform

.. currentmodule:: torch.distributions.uniform

.. autoclass:: Uniform
    :members:
    :undoc-members:
    :show-inheritance:

VonMises

.. currentmodule:: torch.distributions.von_mises

.. autoclass:: VonMises
    :members:
    :undoc-members:
    :show-inheritance:

Weibull

.. currentmodule:: torch.distributions.weibull

.. autoclass:: Weibull
    :members:
    :undoc-members:
    :show-inheritance:

KL Divergence

.. automodule:: torch.distributions.kl

.. currentmodule:: torch.distributions.kl


.. autofunction:: kl_divergence

.. autofunction:: register_kl

Transforms

.. automodule:: torch.distributions.transforms
    :members:
    :member-order: bysource

Constraints

.. automodule:: torch.distributions.constraints
    :members:
    :member-order: bysource

Constraint Registry

.. automodule:: torch.distributions.constraint_registry
    :members:
    :member-order: bysource