sorry for bothering, there is another question.
Now I want to get a new format.left is my input, right is my expected output.
# first element of batch
[[[[1.1,1.2],[1.3,1.4]],                       [[[1.1, 2.1, 3.1, 4.1],
[[2.1,2.2],[2.3,2.4]],                          [[1.2, 2.2, 3.2, 4.2],
[[3.1,3.2],[3.3,3.4]],                          [[1.3, 2.3, 3.3, 4.3],
[[4.1,4.2],[4.3,4.4]]],                         [[1.4, 2.4, 3.4, 4.4]],
 #  second element of batch            
[[[5.1,5.2],[5.3,5.4]],                          [[5.1, 6.1, 7.1, 8.1],
[[6.1,6.2],[6.3,6.4]],                           [5.2, 6.2, 7.2, 8.3],
[[7.1,7.2],[7.3,7.4]],                           [5.3, 6.3, 7.3, 8.3],
[[8.1,8.2],[8.3,8.4]]]]                          [5.4, 6.4, 7.4, 8.4]]]

F.pixel_shuffle is similar but my input size is (N, C*4, H, W) and convert it into (N, C, H*2, W*2) instead of (N, R**2, H, W) into (N, R, H*R, W*R)
relly thank you for your help!