MaxUnpool1d

class torch.nn.modules.pooling.MaxUnpool1d(kernel_size, stride=None, padding=0)

计算 MaxPool1d 的部分逆运算。

MaxPool1d 不是完全可逆的，因为非最大值会被丢失。

要计算 MaxPool1d 的逆运算，需要提供 MaxPool1d 的输出（包括最大值的索引），并计算一个部分逆运算，其中所有非最大值都设置为零。

注意

当输入索引存在重复值时，此操作可能表现出非确定性。更多信息请参阅 pytorch/pytorch#80827 和 可复现性。

注意

MaxPool1d 可以将多种输入大小映射到相同的输出大小。因此，逆运算可能会出现歧义。为了解决这个问题，您可以在前向传播调用中提供所需的输出大小作为附加参数 output_size。请参阅下面的输入和示例。

参数

• kernel_size (int 或 tuple) – 最大池化窗口的大小。

• stride (int 或 tuple) – 最大池化窗口的步幅。默认为 kernel_size。

• padding (int 或 tuple) – 添加到输入的填充。

输入

• input: the input Tensor to invert

• indices: 由 MaxPool1d 给出的索引

• output_size (optional): the targeted output size

形状

• 输入: $(N, C, H_{in})$ 或 $(C, H_{in})$。

• 输出: $(N, C, H_{out})$ 或 $(C, H_{out})$，其中

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

  or as given by output_size in the call operator

示例

>>> pool = nn.MaxPool1d(2, stride=2, return_indices=True)
>>> unpool = nn.MaxUnpool1d(2, stride=2)
>>> input = torch.tensor([[[1., 2, 3, 4, 5, 6, 7, 8]]])
>>> output, indices = pool(input)
>>> unpool(output, indices)
tensor([[[ 0.,  2.,  0.,  4.,  0.,  6.,  0., 8.]]])

>>> # Example showcasing the use of output_size
>>> input = torch.tensor([[[1., 2, 3, 4, 5, 6, 7, 8, 9]]])
>>> output, indices = pool(input)
>>> unpool(output, indices, output_size=input.size())
tensor([[[ 0.,  2.,  0.,  4.,  0.,  6.,  0., 8.,  0.]]])

>>> unpool(output, indices)
tensor([[[ 0.,  2.,  0.,  4.,  0.,  6.,  0., 8.]]])

forward(input, indices, output_size=None)

执行前向传播。

返回类型

张量