torch.nn.functional.fractional_max_pool3d

torch.nn.functional.fractional_max_pool3d(input, kernel_size, output_size=None, output_ratio=None, return_indices=False, _random_samples=None) [источник]

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

Дробный max-пулинг подробно описан в статье Дробный max-пулинг Бена Грэма.

Операция max-пулинга применяется в областях размером $kT \times kH \times kW$ с шагом, определяемым целевым размером выходных данных и выбираемым случайным образом. Количество выходных признаков равно количеству входных плоскостей.

Параметры:

• kernel_size – размер окна, в котором берётся максимум. Может быть одним числом $k$ (для квадратного ядра размером $k \times k \times k$) или кортежем (kT, kH, kW)
• output_size – целевой размер выходных данных в формате $oT \times oH \times oW$. Может быть кортежем (oT, oH, oW) или одним числом $oH$ для кубического выходного тензора размером $oH \times oH \times oH$
• output_ratio – если размер выходных данных нужно задать относительно размера входных данных, можно указать этот параметр. Значением должно быть число или кортеж в диапазоне (0, 1)
• return_indices – если True, возвращает индексы вместе с выходными данными. Это полезно для передачи в max_unpool3d().

Форма:

• Входные данные: $(N, C, T_{in}, H_{in}, W_{in})$ или $(C, T_{in}, H_{in}, W_{in})$.
• Выходные данные: $(N, C, T_{out}, H_{out}, W_{out})$ или $(C, T_{out}, H_{out}, W_{out})$, где $(T_{out}, H_{out}, W_{out})=\text{output\_size}$ или $(T_{out}, H_{out}, W_{out})=\text{output\_ratio} \times (T_{in}, H_{in}, W_{in})$

Примеры::

>>> input = torch.randn(20, 16, 50, 32, 16)
>>> # pool of cubic window of size=3, and target output size 13x12x11
>>> F.fractional_max_pool3d(input, 3, output_size=(13, 12, 11))
>>> # pool of cubic window and target output size being half of input size
>>> F.fractional_max_pool3d(input, 3, output_ratio=(0.5, 0.5, 0.5))