Module: tf.distributions

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)

Other Members

  • FULLY_REPARAMETERIZED
  • NOT_REPARAMETERIZED