layer_kl_divergence_regularizer function

Regularizer that adds a KL divergence penalty to the model loss

tfprobability packageRead PDF manual
  • Maintainer: Tomasz Kalinowski
  • License: Apache License (>= 2.0)
  • Last published: 2025-08-21

Useful links