jax.random.rademacher

jax.random.rademacher(key, shape=(), dtype=<class 'int'>)

從 Rademacher 分佈中取樣。

這些值根據以下機率質量函數分佈

\[f(k) = \frac{1}{2}(\delta(k - 1) + \delta(k + 1))\]

在域 \(k \in \{-1, 1\}\) 上，其中 \(\delta(x)\) 是狄拉克 delta 函數。

參數:

• key (ArrayLike) – PRNG 金鑰。

• shape (Shape) – 返回樣本的形狀。預設為 ()。

• dtype (DTypeLikeInt) – 用於樣本的類型。

返回:

形狀為 shape 的 jnp.array 樣本。輸出中的每個元素有 50% 的機率為 1 或 -1。

返回類型:

Array