TensorFlow

API

 tf.compat / v1 / v1.distribute.experimental.TPUStrategy


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)

FULLY_REPARAMETERIZED tf.compat.v1.distributions.ReparameterizationType
NOT_REPARAMETERIZED tf.compat.v1.distributions.ReparameterizationType

此页内容是否对您有帮助