torch.nn.functional.adaptive_avg_pool2d
torch.nn.functional.adaptive_avg_pool2d(input, output_size)
Applies a 2D adaptive average pooling over an input signal composed
   of several input planes.
See "AdaptiveAvgPool2d" for details and output shape.
Parameters:
      output_size (None) -- the target output size (single
      integer or double-integer tuple)
Return type:
      Tensor