tf.contrib.distributions.kl()

tf.contrib.distributions.kl(dist_a, dist_b, allow_nan=False, name=None)

Get the KL-divergence KL(dist_a || dist_b).

Args:
  • dist_a: The first distribution.
  • dist_b: The second distribution.
  • allow_nan: If False (default), a runtime error is raised if the KL returns NaN values for any batch entry of the given distributions. If True, the KL may return a NaN for the given entry.
  • name: (optional) Name scope to use for created operations.
Returns:

A Tensor with the batchwise KL-divergence between dist_a and dist_b.

Raises:
  • NotImplementedError: If no KL method is defined for distribution types of dist_a and dist_b.
doc_TensorFlow
2016-10-14 12:55:15
Comments
Leave a Comment

Please login to continue.