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