I want to ask if there is a module that has the same function as tf.image.extract_image_patches in Pytorch.
Does anybody know it? Thanks!
It seems tensor.unfold would be the equivalent.
Thanks for your help!
After the experiment, it is indeed equivalent.