dynode.infer.sample.sample_then_resolve

Contents

dynode.infer.sample.sample_then_resolve#

dynode.infer.sample.sample_then_resolve(parameters: Any, rng_key: Array | None = None) → TransmissionParams#

Copy, sample and resolve parameters, returning a jax-compliant copy.

Parameters#

parametersAny: object containing numpyro.Distribution objects to sample and dynode.typing.DeterministicParameter objects to resolve.
rng_keyArray, optional: PRNGKey needed to sample distributions, generated from jax.random.PRNGKey(), by default None meaning context RNGKey will be used if running from within MCMC execution.

Returns#

Any: COPY of the parameters object with all occurences of numpyro.Distribution or DeterministicParameter objects replaced with samples / resolved values.