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 (ArrayLike) – PRNG 键。
shape (Shape) – 返回样本的形状。 默认 ()。
dtype (DTypeLikeInt) – 样本使用的类型。
- 返回值:
一个 jnp.array 类型的样本，形状为 shape。 输出中的每个元素有 50% 的几率是 1 或 -1。
- 返回类型: