jax.random.bernoulli

内容

jax.random.bernoulli#

jax.random.bernoulli(key, p=np.float32(0.5), shape=None)[source]#

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

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

\[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 | None) – 可选，表示结果形状的非负整数元组。必须与 p.shape 广播兼容。默认值（None）产生的结果形状等于 p.shape。

返回值：

如果 shape 不为 None，则返回一个布尔类型且形状由 shape 给定的随机数组；否则返回形状为 p.shape 的数组。

返回类型:

数组

内容