Rate this Page
★★★★★

torch.nn.functional.adaptive_avg_pool2d#

torch.nn.functional.adaptive_avg_pool2d(input,output_size)[source]#

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

SeeAdaptiveAvgPool2d for details and output shape.

Parameters

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

Return type

Tensor