TensorFlow 2.0 Beta is available Learn more

Module: tfp.trainable_distributions

Support for trainable distributions.

Defined in python/trainable_distributions/__init__.py.

Functions

bernoulli(...): Constructs a trainable tfd.Bernoulli distribution.

multivariate_normal_tril(...): Constructs a trainable tfd.MultivariateNormalTriL distribution.

normal(...): Constructs a trainable tfd.Normal distribution.

poisson(...): Constructs a trainable tfd.Poisson distribution.

softplus_and_shift(...): Converts (batch of) scalars to (batch of) positive valued scalars.

tril_with_diag_softplus_and_shift(...): Converts (batch of) vectors to (batch of) lower-triangular scale matrices.