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]¶ (overrides
PerceptionBlock
)implementation of
PerceptionBlock
interface