tf.contrib.bayesflow.variational_inference.register_prior

tf.contrib.bayesflow.variational_inference.register_prior

register_prior(
    variational,
    prior
)

Defined in tensorflow/contrib/bayesflow/python/ops/variational_inference_impl.py.

See the guide: BayesFlow Variational Inference (contrib) > Ops

Associate a variational StochasticTensor with a Distribution prior.

This is a helper function used in conjunction with elbo that allows users to specify the mapping between variational distributions and their priors without having to pass in variational_with_prior explicitly.

Args:

  • variational: StochasticTensor q(Z). Approximating distribution.
  • prior: Distribution p(Z). Prior distribution.

Returns:

None

Raises:

  • ValueError: if variational is not a StochasticTensor or prior is not a Distribution.