Type RegisterKL
Namespace tensorflow.distributions
Parent PythonObjectContainer
Interfaces IRegisterKL
Decorator to register a KL divergence implementation function. Usage: @distributions.RegisterKL(distributions.Normal, distributions.Normal)
def _kl_normal_mvn(norm_a, norm_b):
# Return KL(norm_a || norm_b)
Methods
Properties
Public instance methods
object __call__(string kl_fn)
Perform the KL registration.
Parameters
-
stringkl_fn - The function to use for the KL divergence.
Returns
-
object - kl_fn