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 (ArrayLike) – 用作随机密钥的 PRNG 密钥。
p (RealArray) – 可选,一个浮点数或浮点数数组,表示随机变量的均值。必须与
shape
广播兼容。默认为 0.5。shape (Shape | None | None) – 可选,一个非负整数的元组,表示结果的形状。必须与
p.shape
广播兼容。默认值 (None) 生成与p.shape
相等的结果形状。
- 返回值:
一个具有布尔数据类型,并且形状由
shape
给定的随机数组(如果shape
不是 None),否则形状为p.shape
。- 返回类型: