EngineBuilder.set_initial_values()#
- EngineBuilder.set_initial_values(model_state, multiple_chains=False)[source]#
Sets the initial model state.
If
multiple_chainsis true themodel_statewill be used as is; otherwisemodel_statewill be used as the initial values for each chain. Note that ifmultiple_chainsis true, the first axis of each leaf ofmodel_staterefers to the chain.