torch.nn.functional.avg_pool2d

torch.nn.functional.avg_pool2d(input, kernel_size, stride=None, padding=0, ceil_mode=False, count_include_pad=True, divisor_override=None) → Tensor

Применяет 2D операцию усредняющего пулинга в областях $kH \times kW$ с шагом $sH \times sW$. Количество выходных признаков равно количеству входных каналов.

См. AvgPool2d для получения подробностей и формы выходных данных.

Параметры

• input – входной тензор $(\text{minibatch} , \text{in\_channels} , iH , iW)$
• kernel_size – размер области пулинга. Может быть одним числом или кортежем (kH, kW)
• stride – шаг операции пулинга. Может быть одним числом или кортежем (sH, sW). Значение по умолчанию: kernel_size
• padding – неявные нулевые заполнения по обе стороны от входных данных. Может быть одним числом или кортежем (padH, padW). Значение по умолчанию: 0
• ceil_mode – если True, будет использовать ceil вместо floor в формуле для вычисления формы выходных данных. Значение по умолчанию: False
• count_include_pad – если True, нулевые заполнения будут включены в вычисление среднего. Значение по умолчанию: True
• divisor_override – если указано, оно будет использовано в качестве делителя, иначе будет использоваться размер области пулинга. Значение по умолчанию: None