Distribution Funsors¶
This interface provides a number of standard normalized probability distributions implemented as funsors.
- class Distribution(*args, **kwargs)[source]¶
Bases:
Funsor
Funsor backed by a PyTorch/JAX distribution object.
- Parameters
*args – Distribution-dependent parameters. These can be either funsors or objects that can be coerced to funsors via
to_funsor()
. See derived classes for details.
- dist_class = 'defined by derived classes'¶
- property has_enumerate_support¶
- class Beta(*args, **kwargs)¶
Bases:
Distribution
- class Cauchy(*args, **kwargs)¶
Bases:
Distribution
- class Chi2(*args, **kwargs)¶
Bases:
Distribution
- class BernoulliProbs(*args, **kwargs)¶
Bases:
Distribution
- dist_class¶
alias of
_PyroWrapper_BernoulliProbs
- class BernoulliLogits(*args, **kwargs)¶
Bases:
Distribution
- dist_class¶
alias of
_PyroWrapper_BernoulliLogits
- class Binomial(*args, **kwargs)¶
Bases:
Distribution
- class Categorical(*args, **kwargs)¶
Bases:
Distribution
- dist_class¶
alias of
Categorical
- class CategoricalLogits(*args, **kwargs)¶
Bases:
Distribution
- dist_class¶
alias of
_PyroWrapper_CategoricalLogits
- class Delta(*args, **kwargs)¶
Bases:
Distribution
- class Dirichlet(*args, **kwargs)¶
Bases:
Distribution
- class DirichletMultinomial(*args, **kwargs)¶
Bases:
Distribution
- dist_class¶
alias of
DirichletMultinomial
- class Exponential(*args, **kwargs)¶
Bases:
Distribution
- dist_class¶
alias of
Exponential
- class Gamma(*args, **kwargs)¶
Bases:
Distribution
- class GammaPoisson(*args, **kwargs)¶
Bases:
Distribution
- dist_class¶
alias of
GammaPoisson
- class Geometric(*args, **kwargs)¶
Bases:
Distribution
- class Gumbel(*args, **kwargs)¶
Bases:
Distribution
- class HalfCauchy(*args, **kwargs)¶
Bases:
Distribution
- dist_class¶
alias of
HalfCauchy
- class HalfNormal(*args, **kwargs)¶
Bases:
Distribution
- dist_class¶
alias of
HalfNormal
- class Laplace(*args, **kwargs)¶
Bases:
Distribution
- class Logistic(*args, **kwargs)¶
Bases:
Distribution
- class LowRankMultivariateNormal(*args, **kwargs)¶
Bases:
Distribution
- dist_class¶
alias of
LowRankMultivariateNormal
- class Multinomial(*args, **kwargs)¶
Bases:
Distribution
- dist_class¶
alias of
Multinomial
- class MultivariateNormal(*args, **kwargs)¶
Bases:
Distribution
- dist_class¶
alias of
MultivariateNormal
- class NonreparameterizedBeta(*args, **kwargs)¶
Bases:
Distribution
- dist_class¶
alias of
NonreparameterizedBeta
- class NonreparameterizedDirichlet(*args, **kwargs)¶
Bases:
Distribution
- dist_class¶
alias of
NonreparameterizedDirichlet
- class NonreparameterizedGamma(*args, **kwargs)¶
Bases:
Distribution
- dist_class¶
alias of
NonreparameterizedGamma
- class NonreparameterizedNormal(*args, **kwargs)¶
Bases:
Distribution
- dist_class¶
alias of
NonreparameterizedNormal
- class Normal(*args, **kwargs)¶
Bases:
Distribution
- class Pareto(*args, **kwargs)¶
Bases:
Distribution
- class Poisson(*args, **kwargs)¶
Bases:
Distribution
- class StudentT(*args, **kwargs)¶
Bases:
Distribution
- class Uniform(*args, **kwargs)¶
Bases:
Distribution
- class VonMises(*args, **kwargs)¶
Bases:
Distribution