TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tfp.vi.total_variation

View source on GitHub

The Total Variation Csiszar-function in log-space.

tfp.vi.total_variation(
    logu,
    name=None
)

A Csiszar-function is a member of,

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

The Total-Variation Csiszar-function is:

f(u) = 0.5 |u - 1|

Args:

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

Returns:

  • total_variation_of_u: float-like Tensor of the Csiszar-function evaluated at u = exp(logu).