torch.nn.functional.adaptive_max_pool2d
torch.nn.functional.adaptive_max_pool2d(args, *kwargs)
Applies a 2D adaptive max pooling over an input signal composed of
   several input planes.
See "AdaptiveMaxPool2d" for details and output shape.
Parameters:
      * output_size -- the target output size (single integer or
        double-integer tuple)
  * **return_indices** -- whether to return pooling indices.
    Default: "False"
