ConcatenationBlock

class maze.perception.blocks.general.concat.ConcatenationBlock(*args: Any, **kwargs: Any)

A feature concatenation block.

Parameters

• in_keys – Keys identifying the input tensors.

• out_keys – Keys identifying the output tensors.

• in_shapes – List of input shapes.

• concat_dim – The index of the concatenation dimension.

forward(block_input: Dict[str, torch.Tensor]) → Dict[str, torch.Tensor]

implementation of PerceptionBlock interface