fit_a_nef.initializers.RandomInit

class fit_a_nef.initializers.RandomInit(init_rng: Array)

Initializes the model with different parameters for each signal. The parameters are sampled using different random number generators. The rng is split for each signal starting from the init_rng.

Parameters:

init_rng (jnp.ndarray) – The initial random number generator.

__init__(init_rng: Array)

Constructor method.

Methods

__init__(init_rng)

Constructor method.