Join us at TensorFlow World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

Module: tfp.layers.dense_variational_v2.kullback_leibler

Registration and usage mechanisms for KL-divergences.

Defined in python/distributions/kullback_leibler.py.

Classes

class RegisterKL: Decorator to register a KL divergence implementation function.

Functions

kl_divergence(...): Get the KL-divergence KL(distribution_a || distribution_b).

augment_kl_xent_docs(...): Augments doc on tfd.kl_divergence, EachDist.kl_divergence/cross_entropy.