Hi all, I have an input tensor of shape 12x3x3 which corresponds to 12 patches of size 3x3, can be explained as {Patch1, Patch2, …, Patch12}. I want to convert it to 1x12x9, keeping the patches as described below,
Patch1 Patch2 Patch3
Patch4 Patch5 Patch6
Patch7 Patch8 Patch9
Patch10 Patch11 Patch12
I tried torch.unfold() with different parameters, but couldn’t find the right way. Any ideas?