tf.contrib.distributions.RegisterKL.__call__()

tf.contrib.distributions.RegisterKL.__call__(kl_fn)

Perform the KL registration.

Args:

• kl_fn: The function to use for the KL divergence.

Returns:

kl_fn

Raises:

• TypeError: if kl_fn is not a callable.
• ValueError: if a KL divergence function has already been registered for the given argument classes.