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)

Применяет 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 позже