torch.nn.functional.adaptive_avg_pool2d

torch.nn.functional.adaptive_avg_pool2d(input, output_size)

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

See AdaptiveAvgPool2d for details and output shape.

Parameters

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

Return type

Tensor