jax.scipy.special.rel_entr

jax.scipy.special.rel_entr(p, q)

相对熵函数。

JAX 实现的 scipy.special.rel_entr。

\[\begin{split} \mathrm{rel\_entr}(p, q) = \begin{cases} p\log(p/q) & p>0,q>0\\ 0 & p=0,q\ge 0\\ \infty & \mathrm{否则} \end{cases}\end{split}\]

参数:

• p (类数组) – 类数组，实数值。

• q (类数组) – 类数组，实数值。

返回:

相对熵值的数组。

返回类型:

数组

另请参阅

• jax.scipy.special.entr()

• jax.scipy.special.kl_div()