torch.nn.functional.max_pool1d

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

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

Примечание

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

См. MaxPool1d для подробностей.

Параметры

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