Consider the following 6x6 tensor. tst_tensor = tensor([[1, 1, 1, 2, 3, 1], [1, 2, 0, 0, 0, 0], [1, 0, 1, 0, 0, 0], [5, 5, 5, 5, 5, 5], [5, 5, 5, 0, 0, 0], [0, 2, 3, 0, 0, 0]]) I’d like to window it in 2x2 windows (or 4 x 1) as follows tensor([[1, 1, 1, 2],…

For getting tiles, manually doing this is probably most efficient: windows = tst_tensor.view(3, 2, 3, 2).permute(0, 2, 1, 3).reshape(1, 9, 4) Best regards Thomas