在Pytorch中实现im2col操作 Implementing im2col in Pytorch
在Pytorch中可以用torch.unfold,torch.cat和torch.transpose的组合实现im2col操作.TAKEAWAY:stride=(1,1)kernel_size=(3,3)x=torch.arange(0,25).resize_(5,5)y=torch.cat(torch.cat(x.unfold(0,kernel_size[0],stride[0]).unfold