jax.scipy.special.entr

jax.scipy.special.entr(x)

熵函数

JAX implementation of scipy.special.entr.

\[\begin{split}\mathrm{entr}(x) = \begin{cases} -x\log(x) & x > 0 \\ 0 & x = 0\\ -\infty & x > 0 \end{cases}\end{split}\]

参数:

x (ArrayLike) – 类数组，实值。

返回:

包含熵值的数组。

返回类型:

Array

参见

• jax.scipy.special.kl_div()

• jax.scipy.special.rel_entr()