zahra:

a = torch.tensor([[[ 1, 2, 3, 4, 5, 6], [ 7, 8, 9, 10, 11, 12], [13, 14, 15, 16, 17, 18], [19, 20, 21, 22, 23, 24]]])


Ow, sorry.
I used the tensor in the first post.

a.unfold(2, 2,2).unfold(3, 2,2).contiguous().view(2, 6, 2, 2)

by the way, as I told you only need to work with dim and number of unfold functions. Easy does it.
Bests
Nik