tfp.substrates.jax.vi.chi_square

The chi-Square Csiszar-function in log-space.

View aliases

Main aliases

tfp.experimental.substrates.jax.vi.chi_square

tfp.substrates.jax.vi.chi_square(
    logu, name=None
)

A Csiszar-function is a member of,

F = { f:R_+ to R : f convex }.

The Chi-square Csiszar-function is:

f(u) = u**2 - 1

Warning: this function makes non-log-space calculations and may therefore be numerically unstable for |logu| >> 0.

Args
logu	float-like Tensor representing log(u) from above.
name	Python str name prefixed to Ops created by this function.

Returns
chi_square_of_u	float-like Tensor of the Csiszar-function evaluated at u = exp(logu).