torch.nn.functional.max_pool3d

torch.nn.functional.max_pool3d(input, kernel_size, stride=None, padding=0, dilation=1, ceil_mode=False, return_indices=False) [source]

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

Примечание

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

Подробную информацию см. в MaxPool3d.

Параметры:

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