MaxUnpool3d

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

計算 MaxPool3d 的一個部分逆運算。

MaxPool3d 不是完全可逆的，因為非最大值會被丟失。MaxUnpool3d 以 MaxPool3d 的輸出來作為輸入，包括最大值的索引，並計算一個部分逆運算，其中所有非最大值都被設定為零。

注意

當輸入索引存在重複值時，此操作可能表現出非確定性。更多資訊請參閱 pytorch/pytorch#80827 和 可復現性。

注意

MaxPool3d 可以將多個輸入大小對映到相同的輸出大小。因此，逆運算可能會產生歧義。為了解決這個問題，您可以在 forward 呼叫中提供所需的輸出大小作為附加引數 output_size。請參閱下面的“輸入”部分。

引數

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

• stride (int 或 tuple) – 最大池化視窗的步幅。預設為 kernel_size。

• padding (int 或 tuple) – 新增到輸入的填充。

輸入

• input: the input Tensor to invert

• indices：由 MaxPool3d 輸出的索引

• output_size (optional): the targeted 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]}$$

  or as given by output_size in the call operator

示例

>>> # 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])

forward(input, indices, output_size=None)

執行前向傳播。

返回型別

張量