Module: tfp.vi

Methods and objectives for variational inference.

Modules

mutual_information module: Mutual information estimators and helpers.

Classes

class GradientEstimators: Gradient estimators for variational losses.

Functions

amari_alpha(...): The Amari-alpha Csiszar-function in log-space.

arithmetic_geometric(...): The Arithmetic-Geometric Csiszar-function in log-space.

chi_square(...): The chi-Square Csiszar-function in log-space.

csiszar_vimco(...): Use VIMCO to lower the variance of gradient[csiszar_function(log(Avg(u))]. (deprecated)

dual_csiszar_function(...): Calculates the dual Csiszar-function in log-space.

fit_surrogate_posterior(...): Fit a surrogate posterior to a target (unnormalized) log density.

fit_surrogate_posterior_stateless(...): Fit a surrogate posterior to a target (unnormalized) log density.

jeffreys(...): The Jeffreys Csiszar-function in log-space.

jensen_shannon(...): The Jensen-Shannon Csiszar-function in log-space.

kl_forward(...): The forward Kullback-Leibler Csiszar-function in log-space.

kl_reverse(...): The reverse Kullback-Leibler Csiszar-function in log-space.

log1p_abs(...): The log1p-abs Csiszar-function in log-space.

modified_gan(...): The Modified-GAN Csiszar-function in log-space.

monte_carlo_variational_loss(...): Monte-Carlo approximation of an f-Divergence variational loss. (deprecated arguments)

pearson(...): The Pearson Csiszar-function in log-space.

squared_hellinger(...): The Squared-Hellinger Csiszar-function in log-space.

symmetrized_csiszar_function(...): Symmetrizes a Csiszar-function in log-space.

t_power(...): The T-Power Csiszar-function in log-space.

total_variation(...): The Total Variation Csiszar-function in log-space.

triangular(...): The Triangular Csiszar-function in log-space.