class*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).

  • 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