pytorch手動實現滑動窗口操作,論fold和unfold函數的使用

在卷積網絡中,經常會需要用到卷積核滑動窗口的操作,如下圖所示。這個操作在大多數的深度學習框架中,都被封裝的很好,以至於我們並不需要顯式地調用便可以實現卷積網絡的這個操作。 但是,大部分深度學習框架也是提供了顯式地進行滑動窗口操作的API的,在pytorch中就是unfold和fold。接下來我們來探討下這兩個函數的使用。 在pytorch中,和unfold有關的有:torch.nn.Unfold,
相關文章
相關標籤/搜索