traveler.core.array_tools.dict_concat

traveler.core.array_tools.dict_concat(results)

Concatenate corresponding arrays in nested dictionaries.

Parameters:

results (sequence of dict) – Dictionaries with matching nested structure. Leaves must be JAX arrays or CategoricalArray objects.

Returns:

A dictionary with corresponding leaves concatenated along their first axis.

Return type:

dict