oneflow.nn.functional.adaptive_max_pool3d

oneflow.nn.functional.adaptive_max_pool3d(input, output_size, return_indices: bool = False)

Applies a 3D 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_pool3d.html

See AdaptiveMaxPool3d for details and output shape.

Parameters
  • output_size – the target output size (single integer or triple-integer tuple)

  • return_indices – whether to return pooling indices. Default: False