jax.random.bernoulli
jax.random.bernoulli#
- jax.random.bernoulli(key, p=0.5, shape=None)[source]#
Sample Bernoulli random values with given shape and mean.
- Parameters
key (
Union
[Any
,PRNGKeyArray
]) – a PRNG key used as the random key.p (
Any
) – optional, a float or array of floats for the mean of the random variables. Must be broadcast-compatible withshape
. Default 0.5.shape (
Union
[Sequence
[int
],NamedShape
,None
]) – optional, a tuple of nonnegative integers representing the result shape. Must be broadcast-compatible withp.shape
. The default (None) produces a result shape equal top.shape
.
- Return type
- Returns
A random array with boolean dtype and shape given by
shape
ifshape
is not None, or elsep.shape
.