tfp.math.lambertw

View source on GitHub

Computes Lambert W of z element-wise.

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).

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

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