Module: tfp.layers.dense_variational_v2.kullback_leibler

Registration and usage mechanisms for KL-divergences.

Classes

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

Functions

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

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