Save the date! Google I/O returns May 18-20 Register now

tfp.experimental.nn.losses.kl_divergence_exact

Exact KL Divergence.