stack_numpy_dict_list

class maze.train.utils.train_utils.stack_numpy_dict_list(dict_list: List[Dict[str, numpy.ndarray]])

Stack list of dictionaries holding numpy arrays as values.

Parameters

dict_list – A list of identical dictionaries to be stacked, e.g. [{a: 1}, {a: 2}]

Returns

The list entries as a stacked dictionary, e.g. {a : [1, 2]}