from_distr_init

hypercoil.init.base.from_distr_init(*, shape: Tuple[int], distr: Distribution, key: PRNGKey) → Tensor

Sample a tensor from a specified distribution.

Thin wrapper around numpyro.

Parameters:

shapeTensor

Shape of the tensor to populate or initialise from the specified distribution.

distrDistribution

Numpyro distribution object from which to sample values used to populate the tensor.