Bernoulli

distreqx.distributions.Bernoulli

Bernoulli distribution of shape dims.

Bernoulli distribution with parameter probs, the probability of outcome 1.

__init__(logits: typing.Optional[Array] = None, probs: typing.Optional[Array] = None)

Initializes a Bernoulli distribution.

Arguments:

• logits: Logit transform of the probability of a 1 event (0 otherwise), i.e. probs = sigmoid(logits). Only one of logits or probs can be specified.
• probs: Probability of a 1 event (0 otherwise). Only one of logits or probs can be specified.

---