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

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

Примечание

Порядок 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, вместе с максимальными значениями будет возвращён аргмакс. Полезно для последующего вызова torch.nn.functional.max_unpool2d