MaxUnpool3d

class torch.nn.MaxUnpool3d(kernel_size, stride=None, padding=0) [source]

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

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

Примечание

Эта операция может вести себя недетерминированно, когда входные индексы содержат повторяющиеся значения. См. https://github.com/pytorch/pytorch/issues/80827 и Воспроизводимость для получения дополнительной информации.

Примечание

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

Параметры

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

Входы:

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

Форма:

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

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

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

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

Пример:

>>> # pool of square window of size=3, stride=2
>>> pool = nn.MaxPool3d(3, stride=2, return_indices=True)
>>> unpool = nn.MaxUnpool3d(3, stride=2)
>>> output, indices = pool(torch.randn(20, 16, 51, 33, 15))
>>> unpooled_output = unpool(output, indices)
>>> unpooled_output.size()
torch.Size([20, 16, 51, 33, 15])