jax.random.rademacher

jax.random.rademacher(key, shape=(), dtype=None)

从 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 | None) – 用于样本的类型。

返回:

一个 jnp.array 形式的样本，形状为 shape。输出中的每个元素有 50% 的概率为 1 或 -1。

返回类型:

Array