GlobalAveragePoolingBlock¶
-
class
maze.perception.blocks.general.gap.
GlobalAveragePoolingBlock
(*args: Any, **kwargs: Any)¶ A global average pooling block. The block expects the input tensors to have the from (batch-dim, channel-dim, row-dim, column-dim).
- Parameters
in_keys – One key identifying the input tensors.
out_keys – One key identifying the output tensors.
in_shapes – List of input shapes.
-
normalized_forward
(block_input: Dict[str, torch.Tensor]) → Dict[str, torch.Tensor]¶ (overrides
ShapeNormalizationBlock
)implementation of
ShapeNormalizationBlock
interface