jax.scipy.special.xlogy#

jax.scipy.special.xlogy = <jax._src.custom_derivatives.custom_jvp 对象>[源代码]#

计算 x*log(y),当 x=0 时返回 0。

scipy.special.xlogy 的 JAX 实现。

定义为当 \((x, y) = (0, 0)\) 时返回零,并具有自定义导数规则,以便在此点自动微分定义良好。

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

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

返回值:

包含 xlogy 值的数组。

返回类型:

数组