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的矩陣裏面。
相關文章
相關標籤/搜索