stack_numpy_dict_list

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

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}]

  • expand – If True the values are expended by one dimension at dimension zero.

Returns

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