Defined in tensorflow/_api/v1/distributions/__init__.py.
Core module for TensorFlow distribution objects and helpers.
Classes
class Bernoulli: Bernoulli distribution.
class Beta: Beta distribution.
class Categorical: Categorical distribution.
class Dirichlet: Dirichlet distribution.
class DirichletMultinomial: Dirichlet-Multinomial compound distribution.
class Distribution: A generic probability distribution base class.
class Exponential: Exponential distribution.
class Gamma: Gamma distribution.
class Laplace: The Laplace distribution with location loc and scale parameters.
class Multinomial: Multinomial distribution.
class Normal: The Normal distribution with location loc and scale parameters.
class RegisterKL: Decorator to register a KL divergence implementation function.
class ReparameterizationType: Instances of this class represent how sampling is reparameterized.
class StudentT: Student's t-distribution.
class Uniform: Uniform distribution with low and high parameters.
Functions
kl_divergence(...): Get the KL-divergence KL(distribution_a || distribution_b). (deprecated)