stack_and_flatten_spaces

class maze.perception.perception_utils.stack_and_flatten_spaces(input_tensor_dict: Iterable[Dict[str, torch.Tensor]], observation_spaces_dict: Dict[str | int, gymnasium.spaces.Dict])

Merges an iterable of dictionary spaces (usually observations or actions from subsequent sub-steps) into a single dictionary containing all the items.

If one key is present in multiple elements, all its values will be concatenated in the resulting dictionary.

Parameters:

• input_tensor_dict – Iterable of dictionary spaces (usually observations or actions from subsequent sub-steps).

• observation_spaces_dict – The full observation_spaces dict to use in order to infer the concatenation dimension

Returns:

One flat dictionary, containing all keys and values form the elements of the iterable.