pypomp.PanelParameters.to_jax_array¶
- PanelParameters.to_jax_array(param_names: list[str] | None = None, unit_names: list[str] | None = None, **kwargs) Array[source]¶
Convert parameter values to a JAX array.
- Parameters:
param_names (list of str, optional) – Parameter names in the desired order. If
None(default), returns the array in the canonical parameter order.unit_names (list of str, optional) – Unit names in the desired order. If
None(default), returns the array for all units.**kwargs (dict) – Unused in panel parameters.
- Returns:
JAX array of shape
(n_reps, n_units, n_params).- Return type: