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

Module: tfp.experimental.nn.losses

Loss functions for neural networks.

Functions

compute_extra_loss(...)

kl_divergence_exact(...): Exact KL Divergence.

kl_divergence_monte_carlo(...): Monte Carlo KL Divergence.

negloglik(...): Negative log-likelihood.