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 (ArrayLike) – 一个 PRNG 键。

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

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

返回值:

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

返回类型:

数组