torch.nn.functional.lp_pool2d

torch.nn.functional.lp_pool2d(input, norm_type, kernel_size, stride=None, ceil_mode=False) [исходный код]

Выполняет двумерную пулинг-операцию усреднения степеней над входным сигналом, состоящим из нескольких входных плоскостей.

Если сумма всех входных значений, возведённых в степень p, равна нулю, градиент также устанавливается в ноль.

Если ceil_mode равно True, скользящие окна могут выходить за границы, если они начинаются в левой области дополнения или во входных данных. Скользящие окна, которые начинались бы в правой области дополнения, игнорируются.

Подробности см. в разделе LPPool2d.

Тип возвращаемого значения:

Tensor