jax.random.rademacher

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

从拉德马赫分布中采样。

这些值根据以下概率质量函数分布：

\[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