jax.scipy.special.rel_entr

内容

jax.scipy.special.rel_entr#

jax.scipy.special.rel_entr(p, q)[source]#

相对熵函数。

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{otherwise} \end{cases}\end{split}\]
参数:
  • p (ArrayLike) – 数组类,实值。

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

返回值:

相对熵值的数组。

返回值类型:

Array