Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tfp.math.lambertw

View source on GitHub

Computes Lambert W of z element-wise.

tfp.math.lambertw(
    z, name=None
)

The Lambert W function is the inverse of z = w * tf.exp(w). Lambert W is a complex-valued function, but here it returns only the real part of the image of the function. It also only returns the principal branch (also known as branch 0).

Args:

  • z: A Tensor with type float32 or float64.
  • name: A name for the operation (optional).

Returns:

  • lambertw: The Lambert W function evaluated at z. A Tensor with same shape and same dtype as z.