tensor([[[-1.,  0.,  1., -1.,  0., -1., -1.,  0.,  0.,  2.],
         [ 0.,  0.,  0.,  2.,  0.,  1., -1.,  1., -1.,  1.],
         [ 0., -1.,  0.,  0.,  2.,  0.,  0., -1.,  0.,  0.],
         [ 1.,  1.,  1., -1.,  0.,  2., -1.,  0., -1.,  1.],
         [ 1.,  1.,  1.,  1.,  0.,  1.,  2.,  0., -1.,  1.],
         [ 0.,  0., -1., -1.,  1.,  0.,  0.,  2.,  0.,  0.],
         [ 0.,  1.,  0.,  1.,  0.,  1.,  1.,  0.,  2.,  0.],
         [ 1.,  2.,  1.,  0.,  1., -1., -1.,  0., -1.,  0.],
         [-1., -1.,  2.,  0.,  0., -1., -1.,  1.,  0., -1.],
         [ 2., -1.,  1.,  0.,  0., -1., -1.,  0.,  0.,  1.]],

        [[ 1.,  1.,  0.,  0.,  0.,  0.,  0.,  0.,  1.,  2.],
         [ 0.,  0.,  2.,  0.,  0.,  0.,  0.,  1.,  0.,  0.],
         [ 0.,  0.,  0.,  2.,  0., -1.,  0.,  0.,  1.,  0.],
         [ 0.,  1.,  0.,  1.,  0.,  2.,  1.,  0.,  0.,  0.],
         [ 0.,  0.,  0.,  0.,  0., -1.,  2.,  0.,  0.,  0.],
         [ 0.,  0., -1.,  0., -1.,  0.,  0.,  2.,  0.,  0.],
         [ 0.,  2.,  0.,  0.,  0., -1.,  0.,  0.,  0., -1.],
         [-1.,  0.,  0.,  0.,  2.,  0.,  0.,  1.,  0.,  0.],
         [ 0.,  0.,  0., -1.,  0.,  0.,  0.,  0.,  2., -1.],
         [ 2.,  0.,  0.,  0.,  1.,  0.,  0.,  0.,  0., -1.]]], device='cuda:0')
[8, 1, 3, 6, 5, 0, 2, 9, 7, 4]
[2, 5, 9, 0, 3, 4, 1, 6, 7, 8]
tensor([[8, 1, 3, 6, 5, 0, 2, 9, 7, 4],
        [2, 5, 9, 0, 3, 4, 1, 6, 7, 8]], device='cuda:0')
