Normal Distribution

lerax.distribution.Normal

Bases: AbstractDistreqxWrapper[Float[Array, ' dims']]

Normal distribution.

Attributes:

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

Parameters:

Name	Type	Description	Default
`loc`	`Float[ArrayLike, ' dims']`	The mean of the distribution.	required
`scale`	`Float[ArrayLike, ' dims']`	The standard deviation of the distribution.	required

distribution: distributions.Normal = distributions.Normal(
    loc=loc, scale=scale
)

loc: Float[Array, ' dims']

scale: 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__(
    loc: Float[ArrayLike, " dims"],
    scale: Float[ArrayLike, " dims"],
)