Have a question? Connect with the community at the TensorFlow Forum Visit Forum

tfp.layers.dense_variational_v2.kullback_leibler.augment_kl_xent_docs

Augments doc on tfd.kl_divergence, EachDist.kl_divergence/cross_entropy.