MaxUnpool2d

class torch.nn.modules.pooling.MaxUnpool2d(kernel_size, stride=None, padding=0) [исходный код]

Вычисляет частичное обратное преобразование MaxPool2d.

MaxPool2d не является полностью обратимым, поскольку немаксимальные значения теряются.

MaxUnpool2d принимает на вход выходные данные MaxPool2d, включая индексы максимальных значений, и вычисляет частичное обратное преобразование, при котором все немаксимальные значения устанавливаются в ноль.

Примечание

Эта операция может работать недетерминированно, если входные индексы повторяются. Дополнительные сведения см. в pytorch/pytorch#80827 и в разделе Воспроизводимость.

Примечание

MaxPool2d может отображать несколько размеров входных данных в одни и те же размеры выходных данных. Поэтому обратное преобразование может быть неоднозначным. Чтобы устранить эту неоднозначность, можно указать требуемый размер выходных данных в качестве дополнительного аргумента output_size при вызове прямого прохода. См. разделы «Входные данные» и «Пример» ниже.

Параметры:

• kernel_size (int или tuple) – Размер окна пулинга максимума.
• stride (int или tuple) – Шаг окна пулинга максимума. По умолчанию задается значение kernel_size.
• padding (int или tuple) – Заполнение, добавленное ко входным данным

Входные данные:

• input: входной тензор, для которого выполняется обратное преобразование
• indices: индексы, возвращенные MaxPool2d
• output_size (необязательно): целевой размер выходных данных

Форма:

• Входные данные: $(N, C, H_{in}, W_{in})$ или $(C, H_{in}, W_{in})$.

• Выходные данные: $(N, C, H_{out}, W_{out})$ или $(C, H_{out}, W_{out})$, где

  $$H_{out} = (H_{in} - 1) \times \text{stride[0]} - 2 \times \text{padding[0]} + \text{kernel\_size[0]}$$
  $$W_{out} = (W_{in} - 1) \times \text{stride[1]} - 2 \times \text{padding[1]} + \text{kernel\_size[1]}$$

  или как указано в output_size в операторе вызова

Пример:

>>> pool = nn.MaxPool2d(2, stride=2, return_indices=True)
>>> unpool = nn.MaxUnpool2d(2, stride=2)
>>> input = torch.tensor([[[[ 1.,  2.,  3.,  4.],
                            [ 5.,  6.,  7.,  8.],
                            [ 9., 10., 11., 12.],
                            [13., 14., 15., 16.]]]])
>>> output, indices = pool(input)
>>> unpool(output, indices)
tensor([[[[  0.,   0.,   0.,   0.],
          [  0.,   6.,   0.,   8.],
          [  0.,   0.,   0.,   0.],
          [  0.,  14.,   0.,  16.]]]])
>>> # Now using output_size to resolve an ambiguous size for the inverse
>>> input = torch.tensor([[[[ 1.,  2.,  3.,  4.,  5.],
                            [ 6.,  7.,  8.,  9., 10.],
                            [11., 12., 13., 14., 15.],
                            [16., 17., 18., 19., 20.]]]])
>>> output, indices = pool(input)
>>> # This call will not work without specifying output_size
>>> unpool(output, indices, output_size=input.size())
tensor([[[[ 0.,  0.,  0.,  0.,  0.],
          [ 0.,  7.,  0.,  9.,  0.],
          [ 0.,  0.,  0.,  0.,  0.],
          [ 0., 17.,  0., 19.,  0.]]]])

forward(input, indices, output_size=None) [исходный код]

Выполняет прямой проход.

Тип возвращаемого значения:

Tensor