jax.random.bernoulli

jax.random.bernoulli(key, p=np.float32(0.5), shape=None)

使用给定的形状和均值生成伯努利随机值。

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

\[f(k; p) = p^k(1 - p)^{1 - k}\]

其中 \(k \in \{0, 1\}\) 且 \(0 \le p \le 1\)。

参数:

• key (KeyArrayLike) – 一个用作随机密钥的 PRNG 密钥。

• p (RealArray) – 可选，随机变量的均值，可以是浮点数或浮点数数组。必须与 shape 广播兼容。默认值为 0.5。

• shape (Shape | None) – 可选，一个非负整数元组，表示结果形状。必须与 p.shape 广播兼容。默认值（None）生成一个与 p.shape 相等的结果形状。

返回:

一个布尔数据类型的随机数组，形状由 shape 给出，如果 shape 不是 None，否则为 p.shape。

返回类型:

Array