EngineBuilder.set_initial_values()#
- EngineBuilder.set_initial_values(model_state, multiple_chains=False)[source]#
Sets the initial model state.
If
multiple_chains
is true themodel_state
will be used as is; otherwisemodel_state
will be used as the initial values for each chain. Note that ifmultiple_chains
is true, the first axis of each leaf ofmodel_state
refers to the chain.