torch.nn.functional.avg_pool3d

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

Применяет 3D операцию усредняющего пулинга в областях $kT \times kH \times kW$ с шагом $sT \times sH \times sW$. Количество выходных признаков равно $\lfloor\frac{\text{input planes}}{sT}\rfloor$.

Подробности и форму выходной формы см. в AvgPool3d.

Параметры

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