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

参数:

  • key (ArrayLike) – 一个 PRNG 密钥。

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

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

返回:

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

返回类型:

数组