oneflow.adaptive_avg_pool2d

oneflow.adaptive_avg_pool2d(input, output_size, data_format=None)

Applies a 2D adaptive average pooling over an input signal composed of several input planes.

See oneflow.nn.AdaptiveAvgPool2d

Parameters
  • input – input tensor

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