mlx.core.random.bernoulli#

bernoulli(p: scalar | array = 0.5, shape: Sequence[int] | None = None, key: array | None = None, stream: None | Stream | Device = None) → array#

生成服从伯努利分布的随机值。

值是从参数 p 的伯努利分布中采样的。参数 p 可以是 float 或 array，并且必须可以广播到 shape。

参数:

• p (float 或 array, 可选) – 伯努利分布的参数。默认值: 0.5。

• shape (list(int), 可选) – 输出的形状。默认值: p.shape。

• key (array, 可选) – PRNG 密钥。默认值: None。

返回值:

随机整数数组。

返回类型:

array