jax.scipy.special.xlogy

内容

jax.scipy.special.xlogy#

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

计算 x*log(y),对于 x=0 返回 0。

JAX 实现 scipy.special.xlogy.

这被定义为在 \((x, y) = (0, 0)\) 时返回零,并使用自定义导数规则,以便在这一点上自动微分是明确定义的。

参数:
  • x (ArrayLike) – 类数组,实值。

  • y (ArrayLike) – 类数组,实值。

返回值:

包含 xlogy 值的数组。

返回类型:

数组