jax.scipy.special.xlog1py

Contents

jax.scipy.special.xlog1py#

jax.scipy.special.xlog1py = <jax._src.custom_derivatives.custom_jvp object>[source]#

Compute x*log(1 + y), returning 0 for x=0.

JAX implementation of scipy.special.xlog1py.

This is defined to return 0 when \((x, y) = (0, -1)\), with a custom derivative rule so that automatic differentiation is well-defined at this point.

Parameters:
  • x (jax.typing.ArrayLike) – arraylike, real-valued.

  • y (jax.typing.ArrayLike) – arraylike, real-valued.

Returns:

array containing xlog1py values.

Return type:

Array