avg_pool3d

class torch.ao.nn.quantized.functional.avg_pool3d(input, kernel_size, stride=None, padding=0, ceil_mode=False, count_include_pad=True, divisor_override=None) [source]

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

Примечание

Параметры квантования входных данных передаются на выход.

Параметры:

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