MaxUnpool3d¶
- class torch.nn.MaxUnpool3d(kernel_size, stride=None, padding=0)[源代码]¶
计算
MaxPool3d
的部分逆。MaxPool3d
并非完全可逆,因为非最大值会被丢失。MaxUnpool3d
将MaxPool3d
的输出(包括最大值的索引)作为输入,并计算部分逆,其中所有非最大值都设置为零。注意
当输入索引具有重复值时,此操作可能会表现出不确定性。有关更多信息,请参见 https://github.com/pytorch/pytorch/issues/80827 和 可重复性。
注意
MaxPool3d
可以将多个输入大小映射到相同输出大小。因此,反演过程可能会变得模棱两可。为了解决这个问题,您可以在 forward 调用中提供所需的输出大小作为附加参数output_size
。请参见下面的输入部分。- 参数
- 输入
input:要反转的输入张量
indices:由
MaxPool3d
给出的索引output_size(可选):目标输出大小
- 形状
输入: 或 .
输出: 或 ,其中
或者由调用运算符中的
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])