SliceBlock

class maze.perception.blocks.general.slice.SliceBlock(*args: Any, **kwargs: Any)

A slicing block. This is for example useful for slicing the last time step in recurrent blocks by its index.

Parameters:

  • in_keys – One key identifying the input tensors.

  • out_keys – One key identifying the output tensors.

  • in_shapes – List of input shapes.

  • slice_dim – The dimension to slice from.

  • slice_idx – The index to slice.

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

(overrides PerceptionBlock)

implementation of PerceptionBlock interface