Skip to content

Latest commit

 

History

History
450 lines (356 loc) · 10.2 KB

distributions.rst

File metadata and controls

450 lines (356 loc) · 10.2 KB

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:

InverseGamma

.. currentmodule:: torch.distributions.inverse_gamma
.. autoclass:: InverseGamma
    :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:

Wishart

.. currentmodule:: torch.distributions.wishart
.. autoclass:: Wishart
    :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

.. py:module:: torch.distributions.bernoulli
.. py:module:: torch.distributions.beta
.. py:module:: torch.distributions.binomial
.. py:module:: torch.distributions.categorical
.. py:module:: torch.distributions.cauchy
.. py:module:: torch.distributions.chi2
.. py:module:: torch.distributions.continuous_bernoulli
.. py:module:: torch.distributions.dirichlet
.. py:module:: torch.distributions.distribution
.. py:module:: torch.distributions.exp_family
.. py:module:: torch.distributions.exponential
.. py:module:: torch.distributions.fishersnedecor
.. py:module:: torch.distributions.gamma
.. py:module:: torch.distributions.geometric
.. py:module:: torch.distributions.gumbel
.. py:module:: torch.distributions.half_cauchy
.. py:module:: torch.distributions.half_normal
.. py:module:: torch.distributions.independent
.. py:module:: torch.distributions.inverse_gamma
.. py:module:: torch.distributions.kumaraswamy
.. py:module:: torch.distributions.laplace
.. py:module:: torch.distributions.lkj_cholesky
.. py:module:: torch.distributions.log_normal
.. py:module:: torch.distributions.logistic_normal
.. py:module:: torch.distributions.lowrank_multivariate_normal
.. py:module:: torch.distributions.mixture_same_family
.. py:module:: torch.distributions.multinomial
.. py:module:: torch.distributions.multivariate_normal
.. py:module:: torch.distributions.negative_binomial
.. py:module:: torch.distributions.normal
.. py:module:: torch.distributions.one_hot_categorical
.. py:module:: torch.distributions.pareto
.. py:module:: torch.distributions.poisson
.. py:module:: torch.distributions.relaxed_bernoulli
.. py:module:: torch.distributions.relaxed_categorical
.. py:module:: torch.distributions.studentT
.. py:module:: torch.distributions.transformed_distribution
.. py:module:: torch.distributions.uniform
.. py:module:: torch.distributions.utils
.. py:module:: torch.distributions.von_mises
.. py:module:: torch.distributions.weibull
.. py:module:: torch.distributions.wishart