traveler.core.array_tools

Functions

dict_concat(results)	Concatenate corresponding arrays in nested dictionaries.
dict_to_dimension(d[, keys])	Convert a dictionary of arrays to one array.
dict_to_vector(d[, keys])	Convert a dictionary to a vector.
pytree_concat(results[, axis])	Concatenate matching leaves from a sequence of JAX pytrees.

Classes

IndexedTable(index_array, table[, store])	Indexing of the choosers table
Lookup(in_table, base_table, base_index)	A lookup table that can be used to convert a dictionary to a vector.