How are unfold and fold implemented?

1

I want to refer to fold and unfold in torch and use numpy to implement unfold and fold operations.

When I followed the torch source code, I found: torch/nn/modules/fold.py===>>torch/nn/functional.py===>>torch/nn/_functions/thnn/fold.py
ctx._backend.Im2Col_updateOutput(ctx._backend.library_state,
input, output,
kernel_size[0], kernel_size[1],
dilation[0], dilation[1],
padding[0], padding[1],
stride[0], stride[1])

This seems to be the implemented by C, and I can’t check it.

So I am curious about how to implement the unfold and fold?

When I read other people’s implementation, im2col was performed through multiple for loops sliding

I want to speed up the im2col by matrix multiplication, like AXB where X is the input and A and B is auxiliary matrixs. Hope you can give me some suggestions

1 Like
2

You can find the CPU implementation here and the CUDA implementation here.

2 Likes