jax.nn.log1mexp#
- jax.nn.log1mexp = <jax._src.custom_derivatives.custom_jvp object>[source]#
Numerically stable calculation of \(\log(1 - \exp(-x))\).
This function is undefined for \(x < 0\).
Based on TensorFlow’s implementation.
References
- Parameters:
x (ArrayLike)
- Return type: