oneflow.nn.functional.adaptive_max_pool2d¶
-
oneflow.nn.functional.
adaptive_max_pool2d
(input, output_size, return_indices: bool = False, data_format='channels_first')¶ Applies a 2D adaptive max pooling over an input signal composed of several input planes.
The documentation is referenced from: https://pytorch.org/docs/1.10/generated/torch.nn.functional.adaptive_max_pool2d.html
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