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

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

Примечание

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

Подробности см. в MaxPool1d.

Параметры:

• input – входной тензор формы $(\text{minibatch} , \text{in\_channels} , iW)$, размерность minibatch необязательна.
• 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.