Thank you so much.
And there is a further question:
I have a 8x4x4 torch tensor:
x = torch.Tensor([
[[1, 1, 1, 1],
[1, 1, 1, 1],
[1, 1, 1, 1],
[1, 1, 1, 1]],
[[2, 2, 2, 2],
[2, 2, 2, 2],
[2, 2, 2, 2],
[2, 2, 2, 2]],
[[3, 3, 3, 3],
[3, 3, 3, 3],
[3, 3, 3, 3],
[3, 3, 3, 3]],
[[4, 4, 4, 4],
[4, 4, 4, 4],
[4, 4, 4, 4],
[4, 4, 4, 4]],
[[5, 5, 5, 5],
[5, 5, 5, 5],
[5, 5, 5, 5],
[5, 5, 5, 5]],
[[6, 6, 6, 6],
[6, 6, 6, 6],
[6, 6, 6, 6],
[6, 6, 6, 6]],
[[7, 7, 7, 7],
[7, 7, 7, 7],
[7, 7, 7, 7],
[7, 7, 7, 7]],
[[8, 8, 8, 8],
[8, 8, 8, 8],
[8, 8, 8, 8],
[8, 8, 8, 8]]
])
using your answer it converted to a 2x8x8 tensor as:
tensor([[[1., 2., 1., 2., 1., 2., 1., 2.],
[3., 4., 3., 4., 3., 4., 3., 4.],
[5., 6., 5., 6., 5., 6., 5., 6.],
[7., 8., 7., 8., 7., 8., 7., 8.],
[1., 2., 1., 2., 1., 2., 1., 2.],
[3., 4., 3., 4., 3., 4., 3., 4.],
[5., 6., 5., 6., 5., 6., 5., 6.],
[7., 8., 7., 8., 7., 8., 7., 8.]],
    [[1., 2., 1., 2., 1., 2., 1., 2.],
     [3., 4., 3., 4., 3., 4., 3., 4.],
     [5., 6., 5., 6., 5., 6., 5., 6.],
     [7., 8., 7., 8., 7., 8., 7., 8.],
     [1., 2., 1., 2., 1., 2., 1., 2.],
     [3., 4., 3., 4., 3., 4., 3., 4.],
     [5., 6., 5., 6., 5., 6., 5., 6.],
     [7., 8., 7., 8., 7., 8., 7., 8.]]])

but I want to get a 2x8x8 tensor as:
tensor([[[1., 2., 1., 2., 1., 2., 1., 2.],
[3., 4., 3., 4., 3., 4., 3., 4.],
[1., 2., 1., 2., 1., 2., 1., 2.],
[3., 4., 3., 4., 3., 4., 3., 4.],
[1., 2., 1., 2., 1., 2., 1., 2.],
[3., 4., 3., 4., 3., 4., 3., 4.],
[1., 2., 1., 2., 1., 2., 1., 2.],
[3., 4., 3., 4., 3., 4., 3., 4.]],
    [[5., 6., 5., 6., 5., 6., 5., 6.],
     [7., 8., 7., 8., 7., 8., 7., 8.],
     [5., 6., 5., 6., 5., 6., 5., 6.],
     [7., 8., 7., 8., 7., 8., 7., 8.],
     [5., 6., 5., 6., 5., 6., 5., 6.],
     [7., 8., 7., 8., 7., 8., 7., 8.],
     [5., 6., 5., 6., 5., 6., 5., 6.],
     [7., 8., 7., 8., 7., 8., 7., 8.]]])

Could you help me? Thanks a lot!