TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

Module: tfp.trainable_distributions

View source on GitHub

Support for trainable distributions.


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

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

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

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

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

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