相对熵函数。
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
