Search for some functions in Pytorch

1

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!

2

It seems tensor.unfold would be the equivalent.

3

Thanks for your help!
After the experiment, it is indeed equivalent.