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 [исходный код]

Выполняет операцию двумерного усредняющего пулинга в областях размером $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). Не должно превышать половину эффективного размера ядра, то есть $((kernelSize - 1) * dilation + 1) / 2$. По умолчанию: 0
• ceil_mode – если значение True, для вычисления формы выходных данных в формуле вместо floor будет использоваться ceil. По умолчанию: False
• count_include_pad – если значение True, нулевое дополнение будет учитываться при вычислении среднего значения. По умолчанию: True
• divisor_override – если задано, используется в качестве делителя; в противном случае используется размер области пулинга. По умолчанию: None