jax.random.rademacher#
- jax.random.rademacher(key, shape=(), dtype=<class 'int'>)[source]#
从 Rademacher 分布中采样。
这些值根据概率质量函数分布
\[f(k) = \frac{1}{2}(\delta(k - 1) + \delta(k + 1))\]在域 \(k \in \{-1, 1\}\) 上，其中 \(\delta(x)\) 是狄拉克 delta 函数。
- 参数：
key (KeyArrayLike) – 一个 PRNG 密钥。
shape (Shape) – 返回样本的形状。默认值 ()。
dtype (DTypeLikeInt) – 用于样本的类型。
- 返回值：
一个形状为 shape 的 jnp.array 样本。输出中的每个元素都有 50% 的机会是 1 或 -1。
- 返回类型：