pytorch手动实现滑动窗口操作,论fold和unfold函数的使用

在卷积网络中,经常会需要用到卷积核滑动窗口的操作,如下图所示。这个操作在大多数的深度学习框架中,都被封装的很好,以至于我们并不需要显式地调用便可以实现卷积网络的这个操作。 但是,大部分深度学习框架也是提供了显式地进行滑动窗口操作的API的,在pytorch中就是unfold和fold。接下来我们来探讨下这两个函数的使用。 在pytorch中,和unfold有关的有:torch.nn.Unfold,
相关文章
相关标签/搜索