How to implement the im2col like caffe during convolution？

[chenchr]

Hello. Is there a function in pytorch can do im2col like caffe, in which we can also setup the convolution parameters?

[smth]

This package has an im2col and col2im if you want:

szagoruyko/pyinn