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)\) 是狄拉克δ函数。

参数:

• key (KeyArrayLike) – 一个 PRNG 键。

• shape (Shape) – 返回样本的形状。默认 ()。

• dtype (DTypeLikeInt) – 用于样本的类型。

返回:

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

返回类型:

Array