Hi,
yes this helps alot ! Thank you very much.
But this is not the solution to my problem. I need to change the FFN layer in the Transformer pytorch architecture to accept [B, CH, W, H] to let it keep the 2d Positional encoding. To reshape it, its not enough.
Anyways, thanks for your answer ! I wonder if reshape has the same effect as data.flatten(2)