torch.nn.functional.max_pool2d

torch.nn.functional.max_pool2d(input, kernel_size, stride=None, padding=0, dilation=1, ceil_mode=False, return_indices=False)

Применяет 2D максимальное пулингирование к входному сигналу, состоящему из нескольких входных плоскостей.

Примечание

Порядок ceil_mode и return_indices отличается от того, что наблюдается в MaxPool2d, и изменится в будущей версии.

См. MaxPool2d для получения подробностей.

Параметры

• input – входной тензор $(\text{minibatch} , \text{in\_channels} , iH , iW)$, необязательный размер мини-батча.
• kernel_size – размер области пулингирования. Может быть одиночным числом или кортежем (kH, kW)
• stride – шаг операции пулингирования. Может быть одиночным числом или кортежем (sH, sW). По умолчанию: kernel_size
• padding – Неявное заполнение отрицательной бесконечностью, которое добавляется с обеих сторон, должно быть >= 0 и <= kernel_size / 2.
• dilation – Шаг между элементами в скользящем окне, должен быть > 0.
• ceil_mode – Если True, будет использоваться ceil вместо floor для вычисления формы выходных данных. Это гарантирует, что каждый элемент в входном тензоре покрывается скользящим окном.
• return_indices – Если True, будет возвращен argmax вместе с максимальными значениями. Полезно для torch.nn.functional.max_unpool2d позднее