pub fn bernoulli(p: f64, shape: &[usize], key: Option<&Array>) -> Result<Array>
Booleans of shape that are true with probability p, as MLX: uniform < p.
shape
p
uniform < p