关于手动实现类似卷积的滑动窗口速度极慢的问题

如题,如果手动实现的话一般使用双循环,如下

        for i in range(0, k_matrix.shape[2] - self.ksize[0] + 1, self.stride[0]):
            for j in range(0, k_matrix.shape[3] - self.ksize[1] + 1, self.stride[1]):

这是理解卷积窗口运动的实现方法,但是这种方法运用到实际中时,会使反向传播时间变长,所以应该利用pytorch自带的nn.Unfold()函数来实现类似卷积的滑动窗口操作,这样的话,反向传播时间就会大大减少.