Split image into Non-overlapping sub-blocks

2

You could use tensor.unfold to create the patches and reshape them back as explained in this post.