jax.scipy.special.entr

jax.scipy.special.entr(x)

熵函數

scipy.special.entr 的 JAX 實作。

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

參數:

x (ArrayLike) – 類陣列，實數值。

返回:

包含熵值的陣列。

返回類型:

Array

參見

• jax.scipy.special.kl_div()

• jax.scipy.special.rel_entr()