pytorch中的nn.Unfold()函数和fold(函数详解
文章目录1.nn.Unfold()函数2.nn.Fold()函数1.nn.Unfold()函数描述:pytorch中的nn.Unfold()函数,在图像处理领域,经常需要用到卷积操作,但是有时我们只需要在图片上进行滑动的窗口操作,将图片切割成patch,而不需要进行卷积核和图片值的卷积乘法操作。这是就需要用到nn.Unfold()函数,该函数是从一个batch图片中,提取出滑动的局部区域块,也就是卷积操作中的提取kernel filter对应的滑动窗口。torch.nn.Unfold(kernel_s
原创
2021-04-13 13:35:08 ·
17447 阅读 ·
0 评论