Sorry for coming back with a similar question.
How do I make
tensor([[1., 1., 1., 1., 1., 4., 4., 4., 4., 4.]
[2., 2., 2., 2., 2., 5., 5., 5., 5., 5.],
[3., 3., 3., 3., 3., 6., 6., 6., 6., 6.]])
to
tensor([[1., 1., 1., 1., 1.],
[2., 2., 2., 2., 2.],
[3., 3., 3., 3., 3.],
[4., 4., 4., 4., 4.],
[5., 5., 5., 5., 5.],
[6., 6., 6., 6., 6.]])