1. 程式人生 > >Pytorch,MaxUnpool2d中size操作方法

Pytorch,MaxUnpool2d中size操作方法

    下圖所示為最大值的去池化操作,主要包括三個引數,kernel_size: 卷積核大小(一般為3,即3x3的卷積核), stride:步,還有一個新的size。

    從圖中可以看出,它將維度4x4的去池化結果變為5x5。主要通過排序的方法,將4x4裡面的元素按行展開為(0,0,0,0,0,6,0,8,0,0,0,0,0,14...),然後按照次序放到5x5的矩陣裡面。