jax.scipy.special.xlogy

Contents

jax.scipy.special.xlogy#

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

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

JAX implementation of scipy.special.xlogy.

This is defined to return zero when \((x, y) = (0, 0)\), 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 xlogy values.

Return type:

Array