Bernoulli Distribution

lerax.distribution.Bernoulli

Bases: AbstractMaskableDistribution[Bool[Array, ' dims'], Bool[Array, ' dims']], AbstractDistreqxWrapper[Bool[Array, ' dims']]

Bernoulli distribution.

Attributes:

Name	Type	Description
`distribution`	`distributions.Bernoulli`	The underlying distreqx Bernoulli distribution.

Parameters:

Name	Type	Description	Default
`logits`	`Float[ArrayLike, ' dims'] \| None`	The log-odds of the distribution.	`None`
`probs`	`Float[ArrayLike, ' dims'] \| None`	The probabilities of the distribution.	`None`

distribution: distributions.Bernoulli = (
    distributions.Bernoulli(logits=logits, probs=probs)
)

logits: Float[Array, ' dims']

probs: Float[Array, ' dims']

log_prob(value: SampleType) -> Float[Array, '']

prob(value: SampleType) -> Float[Array, '']

sample(key: Key) -> SampleType

entropy() -> Float[Array, '']

mean() -> SampleType

mode() -> SampleType

sample_and_log_prob(
    key: Key,
) -> tuple[SampleType, Float[Array, ""]]

__init__(
    logits: Float[ArrayLike, " dims"] | None = None,
    probs: Float[ArrayLike, " dims"] | None = None,
)

mask(mask: Bool[Array, ' dims']) -> Bernoulli