MAXUNPOOL使用

向上的阿鹏

已于 2022-02-12 12:07:47 修改

阅读量1k

点赞数

分类专栏：代码文章标签： pytorch 深度学习机器学习

于 2022-02-12 12:06:57 首次发布

本文链接：https://blog.csdn.net/weixin_44543648/article/details/122894718

版权

代码专栏收录该内容

42 篇文章 3 订阅

订阅专栏

将包含最大值索引的输出作为输入，MaxPool2d 并计算部分逆，其中所有非最大值都设置为零。

pool = nn.MaxPool2d(2, stride=2, return_indices=True)
unpool = nn.MaxUnpool2d(2, stride=2)
input = torch.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.]]]])

# specify a different output size than input size
unpool(output, indices, output_size=torch.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.]]]])