tf.compat.v1.distributions.Normal

Decorator to register a KL divergence implementation function.

tf.compat.v1.distributions.RegisterKL(
    dist_cls_a, dist_cls_b
)

Usage:

@distributions.RegisterKL(distributions.Normal, distributions.Normal) def _kl_normal_mvn(norm_a, norm_b): # Return KL(norm_a || norm_b)

Args
`dist_cls_a`	the class of the first argument of the KL divergence.
`dist_cls_b`	the class of the second argument of the KL divergence.

Methods

`call`

View source

__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.

TensorFlow

tf.compat / v1 / v1.distributions.Normal

Usage:

Args

Methods

`call`

TensorFlow

tf

tf.audio

tf.autograph

tf.bitwise

tf.compat

tf.config

tf.data

tf.debugging

tf.distribute

tf.dtypes

tf.errors

tf.estimator

tf.experimental

tf.feature_column

tf.graph_util

tf.image

tf.initializers

tf.io

tf.keras

tf.linalg

tf.lite

tf.lookup

tf.losses

tf.math

tf.metrics

tf.nest

tf.nn

tf.optimizers

tf.quantization

tf.queue

tf.ragged

tf.random

tf.raw_ops

tf.saved_model

tf.sets

tf.signal

tf.sparse

tf.strings

tf.summary

tf.sysconfig

tf.test

tf.tpu

tf.train

tf.version

tf.xla

tf.compat / v1 / v1.distributions.Normal

Usage:

Args

Methods

__call__

`call`