oneflow.nn.MaxUnpool2d

class oneflow.nn.MaxUnpool2d(kernel_size: Union[int, Tuple[int, int]], stride: Optional[Union[int, Tuple[int, int]]] = None, padding: Optional[Union[int, Tuple[int, int]]] = 0)

Computes a partial inverse of MaxPool2d.

MaxPool2d is not fully invertible, since the non-maximal values are lost.

MaxUnpool2d takes in as input the output of MaxPool2d including the indices of the maximal values and computes a partial inverse in which all non-maximal values are set to zero.

The interface is consistent with PyTorch. The documentation is referenced from: https://pytorch.org/docs/1.10/generated/torch.nn.MaxUnpool2d.html.

Note

MaxPool2d can map several input sizes to the same output sizes. Hence, the inversion process can get ambiguous. To accommodate this, you can provide the needed output size as an additional argument output_size in the forward call. See the Inputs and Example below.

Parameters
  • kernel_size (int or tuple) – Size of the max pooling window.

  • stride (int or tuple) – Stride of the max pooling window. It is set to kernel_size by default.

  • padding (int or tuple) – Padding that was added to the input

Inputs:
  • input: the input Tensor to invert

  • indices: the indices given out by MaxPool2d

  • output_size (optional): the targeted output size

Shape:
  • Input: \((N, C, H_{in}, W_{in})\) .

  • Output: \((N, C, H_{out}, W_{out})\), where

    \[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]}\]

    or as given by output_size in the call operator

For example:

>>> import oneflow as flow
>>> pool = flow.nn.MaxPool2d(2, stride=2, return_indices=True)
>>> unpool = flow.nn.MaxUnpool2d(2, stride=2)
>>> input = flow.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.]]]], dtype=oneflow.float32)
>>> # specify a different output size than input size
>>> unpool(output, indices, output_size=flow.Size([1, 1, 5, 5])) 
tensor([[[[ 0.,  0.,  0.,  0.,  0.],
        [ 6.,  0.,  8.,  0.,  0.],
        [ 0.,  0.,  0., 14.,  0.],
        [16.,  0.,  0.,  0.,  0.],
        [ 0.,  0.,  0.,  0.,  0.]]]], dtype=oneflow.float32)

Note

When indices contains elements out of the output_size range, an RuntimeError will be raised on the cpu and an indeterminate result will be calculated on the cuda.