Hi arman,
thanks, this is something I did already, I was to use pytorch’s native unfold and have it done on the fly as transformatinos.
They are not the same size, but above works regardless if use // integer division by patch_size.
thanks for the reply, apologies for overlooking it for so long.