topk.txt

tensor([[[14, 15, 53,  ..., 48, 49,  0],
         [32, 33, 31,  ...,  1, 55,  0],
         [ 2, 14, 13,  ...,  0, 54, 55],
         ...,
         [51, 50, 53,  ...,  6,  4,  7],
         [14, 15,  2,  ..., 50, 48, 55],
         [35, 14, 37,  ...,  8, 48,  0]]], device='cuda:0')
tensor([[[ 2,  1,  0,  ..., 10,  7,  9],
         [45, 44, 43,  ..., 26,  7,  9],
         [45, 53, 44,  ..., 10,  7,  9],
         ...,
         [44, 43, 45,  ..., 25, 26,  9],
         [43, 44, 45,  ..., 25,  7,  9],
         [ 2, 45, 44,  ..., 28, 26,  8]]], device='cuda:0')
tensor([[[41, 42, 43,  ..., 35, 36, 39],
         [41, 42, 43,  ...,  4,  6,  3],
         [41, 42, 43,  ..., 38, 36, 39],
         ...,
         [45, 42, 43,  ...,  8, 10, 17],
         [41, 42, 43,  ..., 38, 36, 39],
         [41, 42, 43,  ..., 25,  8, 36]]], device='cuda:0')
tensor([[[43, 44, 45,  ..., 26, 27,  8],
         [43, 44, 45,  ..., 25, 26,  5],
         [43, 44, 45,  ...,  3, 22,  6],
         ...,
         [43, 44, 42,  ..., 21, 22,  5],
         [43, 44, 42,  ...,  3,  4,  5],
         [42, 43, 41,  ..., 49, 48, 54]]], device='cuda:0')
tensor([[[43, 44, 45,  ..., 55,  3,  4],
         [43, 44, 42,  ...,  4,  5, 54],
         [43, 44, 45,  ...,  4, 54, 55],
         ...,
         [43, 44, 42,  ..., 22,  3,  4],
         [43, 44, 42,  ...,  3, 22,  4],
         [43, 44, 45,  ..., 55,  3,  4]]], device='cuda:0')
tensor([[[43, 44, 45,  ..., 55, 22,  8],
         [43, 44, 42,  ..., 21, 22,  4],
         [43, 44, 45,  ..., 52, 53,  8],
         ...,
         [42, 43, 44,  ..., 53,  4,  6],
         [43, 44, 45,  ..., 55, 22,  3],
         [42, 43, 44,  ..., 15, 21, 22]]], device='cuda:0')
tensor([[[42, 43, 44,  ..., 52, 22, 53],
         [42, 43, 44,  ..., 16,  3,  9],
         [42, 43, 44,  ..., 53, 21,  5],
         ...,
         [41, 42, 43,  ...,  9, 23, 21],
         [44, 41, 42,  ..., 47,  2,  4],
         [41, 42, 43,  ..., 22, 21, 55]]], device='cuda:0')
tensor([[[41, 42, 43,  ..., 50, 51,  8],
         [41, 42, 43,  ..., 51,  4, 52],
         [43, 44, 41,  ...,  0,  3, 22],
         ...,
         [41, 42, 43,  ..., 54, 55,  5],
         [42, 43, 44,  ..., 24, 16,  5],
         [41, 42, 43,  ..., 19, 33, 32]]], device='cuda:0')
tensor([[[41, 42, 43,  ..., 11, 22, 10],
         [43, 44, 45,  ..., 23,  3, 21],
         [43, 44, 42,  ...,  1,  6, 26],
         ...,
         [43, 44, 45,  ..., 25,  9,  2],
         [43, 44, 45,  ..., 54, 22,  1],
         [41, 42, 43,  ..., 21, 46, 48]]], device='cuda:0')
tensor([[[42, 43, 44,  ...,  5,  9, 23],
         [43, 44, 45,  ..., 54, 22,  2],
         [43, 42, 44,  ...,  1,  3, 24],
         ...,
         [41, 42, 43,  ...,  1, 46, 49],
         [41, 42, 43,  ..., 16, 54,  3],
         [43, 44, 42,  ..., 53, 54,  8]]], device='cuda:0')
tensor([[[41, 42, 43,  ..., 15, 50, 51],
         [43, 42, 44,  ..., 22,  8,  4],
         [43, 44, 41,  ..., 15, 47, 51],
         ...,
         [41, 42, 45,  ..., 34, 36, 39],
         [42, 43, 44,  ..., 52, 23, 34],
         [41, 42, 43,  ..., 35, 39, 40]]], device='cuda:0')
tensor([[[43, 42, 44,  ..., 24, 25,  2],
         [41, 42, 43,  ..., 49, 46,  7],
         [41, 42, 43,  ..., 32, 52, 38],
         ...,
         [42, 41, 43,  ..., 50, 47, 53],
         [41, 42, 43,  ..., 30, 50, 31],
         [41, 42, 43,  ..., 22,  2, 53]]], device='cuda:0')
tensor([[[42, 43, 44,  ..., 55, 15, 25],
         [42, 43, 44,  ...,  6, 25, 14],
         [41, 42, 43,  ..., 24, 48, 46],
         ...,
         [43, 44, 45,  ..., 15, 16, 24],
         [41, 42, 43,  ..., 14, 51, 50],
         [43, 44, 42,  ..., 23, 25, 21]]], device='cuda:0')
tensor([[[43, 44, 42,  ..., 23, 25, 21],
         [42, 43, 44,  ..., 26, 47, 25],
         [42, 41, 43,  ..., 16, 19, 18],
         ...,
         [43, 44, 42,  ...,  9, 25, 24],
         [41, 42, 43,  ...,  7, 15,  0],
         [45, 42, 43,  ..., 26, 33, 18]]], device='cuda:0')
tensor([[[42, 43, 44,  ..., 25, 54, 15],
         [41, 42, 43,  ..., 39, 25, 36],
         [41, 42, 43,  ..., 23, 54,  7],
         ...,
         [41, 42, 44,  ..., 46,  4, 17],
         [43, 44, 45,  ..., 18, 53, 14],
         [41, 42, 43,  ..., 40, 46, 26]]], device='cuda:0')
tensor([[[41, 42, 43,  ..., 33, 16, 26],
         [42, 43, 44,  ..., 48, 50, 19],
         [43, 44, 42,  ..., 18, 19, 17],
         ...,
         [44, 43, 45,  ..., 50, 28, 40],
         [42, 43, 44,  ..., 48, 51, 15],
         [41, 42, 43,  ..., 27, 26, 51]]], device='cuda:0')
tensor([[[41, 42, 43,  ..., 47, 27, 34],
         [41, 42, 43,  ..., 22, 30, 53],
         [45, 42, 41,  ..., 35, 55, 32],
         ...,
         [45, 41, 42,  ..., 39, 28, 15],
         [45, 41, 42,  ..., 29, 53, 26],
         [42, 43, 44,  ...,  8,  6,  7]]], device='cuda:0')
tensor([[[42, 43, 44,  ..., 34, 35, 26],
         [42, 43, 44,  ..., 49, 15, 48],
         [41, 42, 43,  ..., 19, 20, 48],
         ...,
         [42, 43, 44,  ..., 15,  6, 27],
         [42, 45, 41,  ..., 39, 46, 35],
         [41, 42, 43,  ..., 38, 37, 15]]], device='cuda:0')
tensor([[[43, 41, 42,  ...,  5, 52,  2],
         [42, 43, 44,  ..., 54, 19, 20],
         [45, 44, 42,  ..., 50, 49, 53],
         ...,
         [42, 43, 44,  ...,  8, 15,  7],
         [43, 44, 45,  ..., 46, 51, 49],
         [45, 41, 42,  ..., 17, 38, 52]]], device='cuda:0')
tensor([[[43, 41, 42,  ...,  9, 46,  4],
         [42, 43, 44,  ..., 54, 14, 20],
         [42, 43, 41,  ..., 40,  4,  5],
         ...,
         [42, 43, 44,  ...,  1, 25, 33],
         [41, 42, 43,  ..., 25,  7, 40],
         [41, 42, 43,  ..., 51, 54, 48]]], device='cuda:0')
tensor([[[45, 41, 42,  ..., 49, 53, 50],
         [45, 44, 41,  ..., 22, 52, 13],
         [41, 42, 43,  ..., 17, 54, 34],
         ...,
         [41, 42, 43,  ...,  1, 30, 29],
         [41, 42, 43,  ..., 25, 22, 47],
         [42, 41, 43,  ..., 27, 51, 46]]], device='cuda:0')
tensor([[[43, 44, 42,  ...,  8,  9, 40],
         [45, 44, 43,  ..., 36, 27, 37],
         [41, 42, 43,  ..., 38, 22, 30],
         ...,
         [45, 41, 42,  ..., 16, 28, 26],
         [41, 42, 43,  ..., 21, 22, 48],
         [45, 41, 42,  ..., 20, 54, 16]]], device='cuda:0')
tensor([[[41, 42, 43,  ..., 36, 23, 24],
         [41, 42, 43,  ..., 49, 38, 16],
         [41, 42, 43,  ..., 20,  5, 25],
         ...,
         [41, 42, 43,  ..., 17, 29, 16],
         [43, 44, 45,  ..., 22, 29, 51],
         [41, 42, 43,  ..., 14, 15, 40]]], device='cuda:0')
tensor([[[41, 42, 45,  ..., 35, 36, 30],
         [31, 30, 32,  ..., 21, 24, 25],
         [41, 42, 43,  ..., 17, 14,  1],
         ...,
         [41, 42, 43,  ..., 25, 26,  3],
         [41, 42, 43,  ..., 15, 30, 22],
         [45, 42, 41,  ..., 18, 20, 19]]], device='cuda:0')
tensor([[[44, 45, 42,  ..., 26, 36, 28],
         [45, 42, 44,  ...,  9, 15, 29],
         [45, 44, 42,  ..., 21, 27, 24],
         ...,
         [43, 42, 44,  ..., 28, 51, 53],
         [45, 44, 43,  ..., 47, 50, 46],
         [43, 42, 44,  ..., 16, 37, 18]]], device='cuda:0')
tensor([[[45, 42, 41, 44, 43,  5,  6,  7,  4,  8,  9, 11, 10,  3, 52, 12, 53,
          51, 54,  2, 13, 55, 18, 20, 19, 27, 17, 28, 25, 26, 46],
         [43, 42, 44, 45, 41, 11,  6,  5, 12,  9, 10,  4,  8, 13,  7, 19, 18,
          15,  3, 17,  2, 27, 29, 14, 30, 28, 34, 36, 47, 48, 16],
         [45, 41, 43, 44, 42, 11, 10, 12,  8,  9,  6,  7, 13,  5,  4, 14, 18,
          19, 15, 32, 17, 21, 31, 20, 33, 34, 30, 35, 24, 36, 16],
         [45, 42, 41, 44, 43, 11,  5, 12,  8,  6, 10,  4,  9,  7, 13, 51, 36,
          48, 49, 47, 35, 19, 34, 20, 50, 18, 27, 32, 33, 26, 25],
         [43, 42, 44, 45, 41,  5,  4,  6,  3,  2,  8,  9,  7, 11, 10, 12, 13,
           1, 17, 14, 18, 15, 19, 16, 40, 38, 20, 46, 39, 37, 47],
         [44, 45, 42, 43, 41,  4,  5,  3,  6,  2, 32, 31, 33, 20, 11, 30, 12,
          19, 13, 37, 17, 36, 29, 38, 34, 18,  9,  8, 14, 35, 10],
         [44, 45, 42, 43, 41,  5,  4,  3,  6, 11,  2, 12, 10,  9, 13,  8,  7,
          14, 27, 18, 20, 30, 19, 31, 29, 28,  1, 21, 22, 25, 24],
         [43, 44, 45, 42, 41,  5,  6,  4,  3,  2,  7,  8,  9, 11, 12, 20, 10,
          13, 19, 21, 22, 23, 14,  1, 15, 18, 26, 52, 17, 49, 50],
         [44, 45, 42, 43, 41,  4,  5,  6,  7,  8, 11,  3, 25, 24, 10,  9, 12,
          23, 26, 27, 13,  2, 21, 14, 18, 15, 17, 20, 19, 22, 48],
         [43, 42, 45, 41, 44,  5,  4,  6, 52,  3,  2, 53, 11, 12, 54, 10,  9,
          55, 13, 30, 49, 19, 31, 29, 20, 51, 46, 18, 48, 28, 38],
         [43, 42, 44, 45, 41,  4,  3,  5,  2,  6,  1, 11, 10,  9, 12,  7,  8,
          13, 14, 15, 17, 18, 21, 50, 20, 16, 19, 51, 23, 22, 24],
         [41, 42, 43, 44, 45,  9, 10, 11,  8,  4,  3, 12,  6,  7,  2,  5, 13,
           1, 27, 28, 26,  0, 29, 38, 40, 14, 25, 30, 39, 37, 36],
         [43, 42, 44, 41, 45,  4, 11,  3, 12,  2, 10,  5, 13,  6,  9, 52,  8,
          14, 18, 19,  1, 20, 50, 49, 51, 15, 53, 17, 31, 46, 21],
         [44, 45, 42, 43, 41,  5,  6,  4,  3,  2, 11,  9, 10,  8, 12,  7, 13,
          19, 14, 18, 17, 20, 36, 38, 34, 37, 35, 26, 25, 15, 16],
         [44, 45, 42, 43, 41, 11,  4,  3, 12, 10,  5, 13,  2,  9,  6, 14,  7,
           8, 15, 29, 28, 27, 18,  1, 17, 30, 31, 19, 20, 26, 25],
         [45, 41, 43, 44, 42,  4,  5,  6,  3,  2, 12,  7, 11,  8, 13,  9, 37,
          38, 36, 10, 46,  1, 39, 14, 49, 51, 50, 35, 15, 52, 32],
         [42, 41, 43, 44, 45,  5,  6,  4,  3,  2,  9, 20, 19, 21, 23, 22, 11,
           8, 12, 10,  7, 13, 14, 18, 17, 26,  1, 27, 25, 28, 15],
         [45, 42, 41, 44, 43,  9, 11,  8, 10,  7, 51, 52, 53, 54, 12, 46, 48,
          55, 49, 47,  6,  5, 50, 13, 38, 27, 34, 28, 36, 29,  4],
         [43, 42, 44, 45, 41, 11, 12, 13, 37, 36, 31, 30, 10, 38, 35, 34, 29,
          32,  9, 33,  5, 28, 39,  4, 14, 27, 18, 46, 40, 49, 19],
         [44, 43, 42, 41, 45, 31, 32, 30, 33, 34, 29, 12, 11, 37, 13, 38, 36,
          35, 10, 14, 17, 39, 18, 19, 28, 20, 25, 24, 16, 23, 21],
         [43, 42, 44, 41, 45,  8, 10,  9, 11, 50, 51, 12,  6, 52, 53,  7, 54,
           5, 13, 25,  4, 18, 17, 19, 24, 26, 27, 23, 14, 48, 16],
         [45, 43, 42, 44, 41, 27,  5,  6, 28, 29, 30,  4, 26,  2,  3,  8,  9,
           7, 11, 31, 34, 36, 35, 37, 12, 25, 10, 38, 13, 24, 52],
         [45, 41, 44, 42, 43,  8,  7,  6,  9,  5,  4, 10, 11, 12, 22, 18, 26,
          20, 25, 21, 19, 51, 48, 49, 24,  3, 47, 52, 46, 53, 13],
         [44, 43, 42, 45, 41, 11, 12, 13, 36, 37, 38, 10, 14, 32, 31, 35, 30,
          39, 33, 34, 17, 20, 18, 19, 29,  4, 40,  5,  3,  9,  2],
         [45, 42, 43, 44, 41,  5,  4,  6,  3,  2, 12, 11, 13, 46, 10,  9, 37,
          38, 39, 36, 47, 31, 14, 49, 40, 32, 48, 30, 35, 50, 33],
         [45, 44, 42, 43, 41,  5,  4, 11,  6,  9, 53, 12, 10,  3, 50, 52,  8,
          54, 51, 13,  2,  7, 49, 46, 48, 14, 47, 55, 15, 38, 37],
         [45, 44, 42, 43, 41,  8, 11,  9, 25, 10, 24, 12, 51,  6, 48,  7, 23,
           5, 21, 50, 47,  4, 49, 22, 52, 46, 27, 18, 13, 53, 26],
         [44, 45, 42, 43, 41,  5,  4,  6,  3,  2,  9,  8, 11,  7, 12, 10, 13,
           1, 14, 20, 19, 17, 18, 15, 30, 31, 37, 32, 38, 21, 36],
         [42, 41, 43, 44, 45, 11, 12, 30, 31,  9, 10, 32, 38, 37, 36, 13, 33,
          34, 39, 40, 29, 20, 19,  8, 35, 18, 28, 17,  5,  6, 27],
         [43, 42, 44, 45, 41, 11, 12, 13,  4,  5,  3,  2, 10,  6,  9, 31, 32,
          30, 37, 14, 38, 36, 33, 35, 29, 34, 39, 46, 53, 15, 52],
         [44, 45, 42, 43, 41,  5,  6,  4,  8, 12, 11,  9,  7,  3, 13,  2, 10,
          14, 19, 26, 15, 18, 17, 27, 25, 20, 16, 28, 24, 29, 23],
         [42, 43, 44, 45, 41,  5,  6,  4,  9, 11,  8,  7,  3, 10,  2, 12, 13,
          26, 25, 27, 24, 20, 19, 28, 17, 18, 23, 14, 21, 29, 36]]],
       device='cuda:0')
tensor([[[45, 41, 43, 44, 42,  5,  4,  3,  6,  2, 11, 12, 10, 13,  9,  7, 14,
           8,  1, 31, 38, 15, 30, 37, 32, 40, 18, 19, 39, 46],
         [45, 41, 43, 44, 42,  5,  4,  6,  3,  2,  8, 10,  9, 11, 12,  7, 50,
          51, 29, 52, 38, 53, 31, 30, 13, 37, 28, 39, 36, 54],
         [45, 42, 43, 44, 41, 11, 10, 12,  4, 13,  6,  5,  9,  3,  2, 14,  8,
           7, 17, 37, 31, 38, 18, 36, 20, 30, 32, 19, 39, 16],
         [45, 41, 42, 43, 44,  5,  4,  3,  2,  1, 28, 27, 29, 36, 34, 26, 35,
          25, 37, 38,  9,  8, 11, 30, 12, 10, 31,  6,  7, 32],
         [42, 43, 41, 44, 45, 31, 30, 32, 33, 13, 12, 11, 29, 34, 10, 14,  5,
           4,  3,  9,  2, 36, 37, 35,  6, 17, 28, 18, 15, 16],
         [43, 42, 44, 45, 41,  5,  4,  6,  8,  9, 11, 12, 10,  7,  3, 13,  2,
          14, 15, 17, 16, 18, 19, 20, 31, 30,  1, 29, 46, 21],
         [43, 42, 44, 45, 41, 32, 31, 11, 30, 12, 13, 37, 38, 33, 36, 10, 39,
          14, 29, 35, 34, 19, 20, 18, 17, 40,  9, 16, 15, 28],
         [45, 41, 43, 44, 42,  5,  6,  4, 11, 12, 13,  9,  3,  2, 10,  7,  8,
          31, 14, 32, 30, 19, 20, 33, 18, 29, 37, 36, 34, 17],
         [43, 42, 44, 45, 41, 11, 12, 10, 13,  4,  5,  9,  6, 38,  8,  3, 14,
          30, 37, 46, 18, 31, 36, 21, 24, 25, 19,  2, 17, 39],
         [45, 41, 43, 44, 42, 31, 30, 32, 33, 11, 12, 13,  5,  4, 29,  2,  3,
           6, 37, 38, 14, 34, 36, 10, 39, 35,  9, 20, 15, 40],
         [45, 42, 43, 44, 41,  5,  6,  4,  3,  2, 11, 12, 13, 30, 10, 46, 31,
           9, 37, 32, 38, 36, 29, 34, 33, 49, 35, 48, 47, 20],
         [45, 41, 43, 44, 42,  5, 11,  4, 10, 12,  9,  6,  8,  3, 13,  2,  7,
          14, 15,  1, 17, 16, 18, 19, 20, 50, 49, 51, 47, 48],
         [43, 45, 44, 41, 42,  6,  5,  4,  3,  2,  9,  8,  7, 10, 11, 12, 13,
          18, 19, 20, 14,  1, 17, 21, 22, 26, 15, 31, 29, 27],
         [45, 41, 43, 44, 42, 31, 30, 32, 29, 33, 19, 18, 17, 20, 16, 12, 11,
          13, 10, 14, 28, 34,  9, 36, 37, 35, 15, 38, 24, 23],
         [43, 44, 41, 45, 42, 11, 14, 12, 13, 15, 10, 31, 32, 17, 18, 16, 33,
          30,  9, 34, 20, 29, 35, 37, 38,  5, 24, 21, 23, 39],
         [45, 44, 43, 42, 41, 11, 12, 18, 20, 13, 38, 14, 31, 30, 19, 32, 21,
          39, 46, 22, 10,  4, 33, 37,  3,  5, 36, 29,  2, 15],
         [45, 43, 44, 42, 41,  5,  4, 11, 10, 12,  9,  3,  2,  6,  8, 48, 47,
          13, 25, 24, 50, 49, 23,  7, 51, 27, 18, 26, 17, 22],
         [43, 42, 44, 45, 41,  5,  4,  3,  6,  2, 11, 10, 12,  9, 13,  7,  8,
          18, 14, 20, 31, 19, 17, 37, 29, 30, 36, 32, 28, 38],
         [42, 43, 45, 41, 44, 37, 36, 38, 35, 31, 30, 32, 34, 33, 39, 29, 11,
          12, 10, 13, 40,  5,  4,  6,  3, 14,  2,  9, 28, 18],
         [ 5, 43, 42, 44, 45, 41,  4,  3,  6,  2,  9,  8,  7, 11, 13, 10, 12,
           1, 14, 15, 17, 19, 20, 16, 37, 18, 46, 36, 30, 31],
         [45, 42, 43, 44, 41, 31, 32, 30, 37,  5, 33, 11, 12, 36,  6, 13, 38,
           4,  3,  2, 34, 10, 35, 29, 39,  9, 40, 20, 19, 14],
         [45, 44, 41, 42, 43, 48, 47, 46,  8,  9, 10, 49,  6,  5, 11,  7, 12,
           4, 25, 51, 24, 52,  3, 23, 21,  2, 38, 50, 53, 26],
         [45, 41, 43, 44, 42, 31, 30, 32, 33, 29, 12, 11, 13, 34, 10, 14, 19,
          18, 17, 16, 20, 28, 36, 37, 15, 35, 38,  9, 39, 24],
         [45, 41, 42, 43, 44,  8,  7,  5,  6,  4,  9, 11, 10, 12, 18, 17, 19,
          13, 20, 25, 24, 27, 28, 26, 16, 29, 21, 31, 23, 30],
         [45, 44, 42, 43, 41,  5,  6,  4,  2,  3, 11, 12,  9, 10,  8, 13,  7,
          14, 15, 20, 52, 19, 51, 53, 18, 25, 17, 26, 21, 54],
         [45, 44, 41, 42, 43,  6,  8,  7,  5,  9,  4, 10, 11, 12,  3, 48, 18,
          47, 19, 13,  2, 49, 21, 50, 22, 46, 24, 15, 25, 17],
         [42, 41, 45, 44, 43,  8,  7,  9,  6, 10, 11,  5,  4, 38, 12, 40, 39,
          48, 46, 47, 37, 36, 17, 49, 34, 18, 35, 19, 25, 13],
         [41, 42, 45, 43, 44,  8,  6,  5,  7,  9, 12,  4, 10, 11, 53, 54, 55,
          51, 52, 13, 14, 16, 15, 26, 25, 24, 27, 23,  3,  2],
         [44, 43, 45, 41, 42,  8,  6,  5,  4,  7,  9, 10, 11, 12,  3,  2, 14,
          13, 15, 50, 51, 52, 49, 53, 47, 20, 46, 16, 48, 18],
         [45, 42, 43, 44, 41,  5,  6,  7,  8,  9,  4,  3,  2, 38, 46, 34, 32,
          33, 37, 31, 49, 39, 48, 35, 36, 40, 47, 30, 10, 53],
         [44, 45, 43, 42, 41, 12, 11, 13, 10, 14, 18, 17, 31, 16, 30,  9, 32,
          19,  5, 24, 20, 27, 25, 15, 23, 29, 26,  6,  4, 33],
         [45, 41, 43, 44, 42,  5,  6,  4, 31, 30, 32, 38,  3, 37, 33, 47, 36,
          34, 11,  2, 12, 49, 48,  8,  7, 50, 35, 13, 29, 10]]],
       device='cuda:0')
tensor([[[43, 45, 44, 42, 41,  5,  4,  8,  3,  2,  6,  9,  7, 10,  1, 11, 12,
          50, 53, 51, 54, 52, 13, 48, 47, 14, 46, 49, 18],
         [45, 44, 42, 43, 41,  6,  5,  8,  7,  4, 50, 53, 51,  9, 54, 25, 10,
          24, 11, 52, 23, 26, 12, 27, 48, 47, 19, 49, 18],
         [41, 43, 42, 44, 45,  5,  4,  3,  2,  6, 32, 31, 30, 33, 34, 37, 38,
          29, 36, 19, 20, 35, 39, 12, 13, 18, 40, 11, 21],
         [43, 42, 44, 45, 41,  5,  4,  3,  6,  2, 11, 12, 10,  9, 13,  8,  7,
          14,  1, 18, 17, 19, 31, 30, 20, 16, 29, 15, 28],
         [44, 43, 45, 42, 41,  6,  5,  4,  2,  3, 38, 37, 36,  9, 35, 46,  8,
          39, 40, 34, 10, 31, 11, 32, 30,  7, 33, 29, 47],
         [43, 42, 44, 45, 41,  5,  4,  6, 11,  3, 12,  2, 10,  8,  7, 13,  9,
          14, 18, 17, 19, 20, 15,  1, 16, 31, 21, 24, 30],
         [43, 42, 44, 45, 41, 31, 30, 32, 33, 29, 34, 37, 36, 11,  4, 38, 12,
          13,  5,  3, 10,  2, 35,  6, 28, 39, 14, 19,  9],
         [43, 42, 44, 41, 45,  5,  6,  4,  3,  2, 20, 19, 31, 30, 32, 33, 34,
          18, 17, 21, 36, 37, 35, 12, 14, 13, 38,  7, 22],
         [44, 43, 45, 42, 41, 11, 13,  5, 12, 10,  9,  4, 14,  6,  3,  8, 17,
           2, 31, 32, 15, 30,  7, 16, 18, 19, 33, 29, 20],
         [43, 42, 44, 45, 41, 37, 36, 38, 30, 31, 35, 32, 34, 33, 29, 39,  5,
           6,  4, 28,  3,  2, 40, 11,  9, 27, 12, 10, 13],
         [44, 43, 45, 42, 41,  4,  5,  3,  2,  6,  1, 11, 12, 10,  9, 13, 25,
           8,  7, 24, 14, 23, 18, 27, 26, 17, 50, 19, 15],
         [44, 43, 45, 42, 41,  5,  4,  3,  6,  2,  8,  7,  9, 11,  1, 10, 12,
          13, 14, 15, 17, 16, 46, 38, 18, 20, 19, 37, 36],
         [41, 43, 42, 44, 45, 31, 30, 32, 33, 29, 12, 34, 11, 13, 14, 18, 20,
          17, 28, 19, 10, 35, 36, 16, 37, 38, 15, 21,  9],
         [45, 44, 42, 43, 41,  4,  5,  3,  2,  6,  1, 10, 12, 11,  9,  7, 13,
           8, 14, 15, 17, 18, 19, 16, 20, 50, 49, 27, 51],
         [43, 42, 45, 41, 44,  5,  8,  9,  4,  6, 11, 10, 12,  2,  3,  7, 13,
          53, 29, 27, 54, 50,  1, 28, 26, 52, 51, 46, 30],
         [44, 43, 42, 45, 41, 37, 38, 39, 36, 40, 32, 31, 46, 35, 33, 30, 34,
           6, 20,  5, 29, 11, 13, 12, 19, 10,  4,  9,  3],
         [43, 44, 42, 45, 41, 52, 49, 50, 53, 32, 31, 30, 51, 33, 37, 36, 34,
          38, 35,  5,  6, 39, 29,  4,  3,  2, 13, 40, 14],
         [43, 44, 45, 42, 41, 31, 30, 32, 33, 29, 12, 11, 13, 34, 14, 28, 10,
          18, 19, 17, 20, 16, 15, 37, 36, 35, 38, 27, 21],
         [45, 43, 42, 44, 41,  5,  6,  4,  3,  2,  7,  9,  8, 10, 46, 49, 47,
          11, 48, 12, 51, 35, 50, 34, 13, 36, 52, 32, 31],
         [43, 42, 44, 41, 45, 11, 10,  8,  9,  7,  5,  4,  6,  3, 12,  2, 13,
          25, 24, 52, 50, 53, 51, 21, 23, 22, 14, 18, 54],
         [43, 42, 44, 45, 41,  8,  6,  7,  9,  5, 10,  4, 53, 40, 54, 38, 11,
          52, 46, 39, 51, 37, 33, 34,  3, 32,  2, 36, 31],
         [45, 44, 42, 43, 41, 11,  4,  5, 10,  3, 12,  2, 13,  9,  6,  8,  7,
          14, 15, 17, 18, 16,  1, 19, 20, 46, 38, 36, 32],
         [45, 44, 42, 43, 41,  5,  4,  3,  2,  6, 11, 12, 38, 37,  8, 13,  1,
           7, 36, 10,  9, 40, 39, 35, 46, 34, 14, 31, 19],
         [44, 43, 45, 42, 41,  5,  4,  3,  2,  6,  9,  8, 11, 12, 10,  7,  1,
          13, 14, 15, 17, 16, 18, 20, 38, 19, 52, 21, 53],
         [44, 43, 45, 42, 41,  6,  5,  4,  3,  2,  9,  7,  8, 10, 11, 12, 13,
          31, 19, 37, 30, 34, 32, 36, 33, 18, 20, 38, 35],
         [44, 43, 45, 42, 41,  5,  6,  4,  2,  3, 11, 12, 13,  9, 10,  8, 14,
           7, 19, 20, 17, 15, 18, 38, 37, 31, 46, 21, 16],
         [43, 42, 44, 45, 41, 31, 32, 30, 33, 12, 11,  5, 13,  4, 38, 37,  2,
           3, 34, 29, 36,  6, 10, 35, 39, 14, 40,  9, 28],
         [42, 43, 44, 45, 41, 11, 12, 13,  5, 52,  4, 53, 14, 19,  6,  3, 10,
          54,  2, 20, 46, 55, 18, 17, 23, 21, 32, 31, 22],
         [43, 42, 44, 45, 41,  5,  4,  6,  2,  3, 11,  9, 10, 12, 13,  8,  7,
          37, 36, 31, 30, 33, 14, 34, 35, 32,  1, 29, 52],
         [44, 43, 45, 42, 41, 11, 12, 46,  4, 10, 13,  3,  2, 38, 31, 37, 32,
          48, 39, 49, 47, 36, 14,  5, 50,  6, 33,  9, 30],
         [43, 42, 45, 44, 41, 38, 37, 36,  5, 46,  4, 39, 40,  3, 53, 35,  2,
           6, 34, 54, 32, 52, 33, 31, 12, 30, 51, 49, 55],
         [43, 42, 44, 45, 41, 31, 30, 32, 29, 33, 12, 13, 11, 28, 34, 10, 36,
          37, 35, 27, 38, 19, 18, 20, 14, 17, 16,  5, 26]]], device='cuda:0')
tensor([[[44, 43, 45, 42, 41, 11,  5, 12,  9, 10,  6, 13,  4,  3,  2,  8,  7,
          14, 37, 38, 31, 36, 32, 35, 33, 34, 30, 39],
         [44, 43, 45, 42, 41,  5,  4,  6,  3,  2,  8,  7,  9, 10, 11, 12, 13,
           1, 14, 15, 16, 17, 18, 19, 20, 21, 22, 24],
         [43, 42, 44, 45, 41, 30, 32, 31, 36, 37, 33, 38, 46, 34, 35, 11, 39,
          29, 13, 12, 10, 48, 17, 47,  5, 18, 40, 19],
         [45, 44, 42, 43, 41, 30, 31, 32, 29, 33, 36, 37, 34,  5, 38,  4, 35,
          28, 11,  3, 12, 49, 39, 19, 20, 13,  2, 46],
         [45, 44, 42, 43, 41,  5,  4,  6,  3,  2, 50,  8, 51,  7, 53, 52, 54,
           9, 10, 11, 12,  1, 25, 13, 48, 47, 24, 46],
         [45, 44, 42, 43, 41, 31, 32, 30, 33, 29, 34, 38, 37, 36, 35, 11, 12,
          39,  5, 13, 10,  4,  6, 28, 40, 47, 19, 50],
         [44, 41, 42, 43, 45, 11, 12, 10, 13,  9, 18, 19, 17, 40,  6, 20,  5,
           4, 38, 37,  8, 14, 39, 46,  7, 16, 36, 25],
         [42, 41, 44, 43, 45,  6,  5,  8,  4,  9,  7, 10, 11,  3,  2, 12, 20,
          18, 13, 14, 21, 32, 17, 31, 19, 22, 15, 33],
         [45, 44, 42, 43, 41,  5,  4,  6,  3,  2,  7,  8,  1,  9, 11, 12, 53,
          10, 54, 13, 48, 47, 55, 49, 50, 14, 18, 51],
         [43, 45, 44, 42, 41,  2,  5,  6,  3,  4, 11, 12, 13,  9, 14,  1,  8,
          10, 15,  0, 20, 28, 26,  7, 19, 27, 25, 17],
         [44, 43, 45, 42, 41, 32, 31, 30, 33, 29, 34, 19, 18, 20, 17, 12, 11,
          13, 37, 35, 36, 16, 38, 14, 10, 39, 15, 28],
         [44, 43, 45, 42, 41,  5,  4, 11, 12,  2,  3, 13, 10, 31, 30, 32,  6,
          20, 19, 17, 18, 33,  9, 14, 37, 38, 36, 29],
         [45, 41, 43, 44, 42,  5,  4,  3,  2,  6,  8,  9,  1,  7, 11, 10, 12,
          13, 14, 15, 17, 16, 18, 19, 53, 20, 52, 49],
         [ 9, 43, 42, 44, 41, 45,  8,  6, 19,  5,  7, 12, 20, 11, 10, 13, 18,
          21, 22, 16, 17, 15, 14, 23,  4, 40, 46, 37],
         [44, 43, 45, 42, 41,  5,  2,  4,  6,  3,  8,  7,  1,  9, 11, 10, 12,
           0, 19, 20, 13, 18, 14, 16, 30, 21, 32, 31],
         [44, 45, 42, 43, 41,  5,  6,  8,  4,  7,  9,  3,  2, 10, 12, 11, 49,
          50, 13, 48, 46,  1, 47, 51, 53, 52, 40, 54],
         [44, 43, 45, 42, 41,  5,  4,  3,  6,  2, 12, 11, 30, 31, 32, 10, 29,
          33, 13, 34,  9,  8, 28, 36, 35, 37, 14, 19],
         [44, 43, 45, 42, 41,  5,  4,  6,  3,  2, 11,  8, 12,  9, 10, 13,  7,
          14, 20, 18, 17,  1, 19, 15, 16, 21, 32, 27],
         [43, 42, 44, 45, 41,  5,  4,  6, 10,  3, 11,  7,  8,  9, 12,  2, 13,
          18, 17, 16, 19, 53, 15, 54, 14,  1, 50, 20],
         [44, 43, 41, 42, 45,  4,  5,  6,  3,  2,  8,  7, 11, 12,  9, 10,  1,
          13, 14, 46, 52, 50, 51, 49, 48, 15, 38, 39],
         [43, 42, 44, 45, 41,  9, 11, 10, 12, 13,  8,  7, 31, 32, 30, 18, 20,
           6, 19, 17, 37, 33, 38,  5, 36, 34, 14, 39],
         [45, 44, 42, 43, 41, 11, 10, 12,  9,  8,  5,  6, 51, 50,  4, 13,  7,
          31, 52, 38, 48, 30, 37, 36, 49, 53, 47,  2],
         [44, 43, 45, 42, 41,  6, 30, 19, 31,  5, 18, 32, 17, 16, 33, 37,  8,
          29, 34, 36, 38,  7, 35,  9, 15,  4, 10, 11],
         [43, 42, 44, 45, 41, 32, 31, 30, 33, 11, 13, 12, 29, 20, 14, 34, 17,
          18, 10, 19,  2, 16,  3,  5, 35,  4, 36, 37],
         [44, 43, 45, 42, 41, 11, 12,  5,  4, 13, 10,  3,  2,  6,  9, 14, 32,
          31, 30, 18,  8, 20, 19, 33, 36, 17, 34, 15],
         [43, 42, 44, 45, 41, 31, 32, 30, 33, 38, 13, 12, 37, 36, 14, 29, 11,
          39, 34, 35, 10, 16, 15, 40, 17, 18, 20,  4],
         [44, 43, 45, 42, 41, 31, 32, 30, 33, 29, 34, 11, 12, 13, 28, 14, 36,
          17, 19, 18, 20, 37, 10, 35, 38, 16, 39, 46],
         [44, 43, 45, 42, 41,  5,  4,  3,  2, 11, 12, 10,  9,  1, 13,  6,  8,
          17, 14, 25, 18, 15, 24,  7, 23, 22, 21, 19],
         [44, 43, 42, 45, 41,  5,  4,  3,  2,  6, 13,  9, 36,  1, 11, 34,  8,
          35, 12,  7, 32, 37, 33, 14, 38, 31, 17, 30],
         [45, 44, 42, 43, 41,  5,  4,  3,  6,  2, 11, 12, 13, 10, 14,  9, 17,
          18, 20, 15, 16,  8,  7, 46, 19, 32, 31, 37],
         [43, 45, 44, 42, 41,  5,  4,  3,  2,  6, 11, 12, 13, 37, 36, 38, 46,
          39, 35, 32, 34, 33, 14, 31, 10, 40, 52, 30],
         [44, 43, 45, 42, 41,  5,  6,  4,  3,  2,  9,  7, 11,  8, 10, 13, 12,
          14,  1, 15, 19, 17, 18, 16, 20, 21, 22, 23]]], device='cuda:0')
tensor([[[45, 44, 42, 43, 41,  2,  1,  0,  3,  5,  4,  6,  8,  7,  9, 10, 11,
          12, 13, 19, 14, 17, 18, 15, 20, 16, 21],
         [43, 45, 44, 42, 41,  2,  1,  0,  3,  5,  4,  6, 53, 54,  7,  8, 52,
          50, 51, 11,  9, 12, 10, 55, 13, 49, 14],
         [44, 43, 45, 42, 41,  2,  5,  3,  4,  6,  1, 12, 38,  0, 37, 30, 36,
          11, 13, 29,  9, 27, 31, 35, 28, 10, 34],
         [43, 42, 44, 41, 45,  2,  1,  0,  3,  5,  6,  4,  8,  7,  9, 10, 11,
          12, 13, 14, 15, 48, 17, 16, 47, 18, 19],
         [43, 42, 45, 41, 44,  2,  1,  0,  3, 38,  9,  6, 10, 52, 37, 50, 51,
          36,  8, 39, 31,  5, 35, 34, 11, 53, 32],
         [44, 41, 43, 45, 42, 36, 37, 38, 32, 31, 11, 39, 33, 35, 10, 34, 30,
          12, 13, 18, 26,  9, 19, 25, 27, 17, 29],
         [45, 44, 42, 43, 41,  2,  1,  0,  3,  5,  4, 11, 10,  8,  9,  7,  6,
          12, 13, 14, 15, 18, 17, 16, 20, 19, 30],
         [45, 44, 42, 43, 41,  2,  1,  0,  5,  4,  6,  3,  8,  9,  7, 10, 11,
          12, 13, 38, 37, 15, 36, 35, 19, 14, 18],
         [44, 45, 42, 43, 41,  2,  1,  0,  3, 48, 47, 50,  8, 46, 51, 49,  6,
           5,  4, 10,  9, 11,  7, 12, 52, 53, 54],
         [45, 44, 42, 43, 41,  8,  9, 10,  7, 11,  6, 12, 50, 51,  5, 48, 49,
          13, 47, 52, 46,  4, 53, 54, 25, 18, 19],
         [42, 43, 41, 44, 45, 31, 32, 30, 33, 29, 36, 37,  2, 38, 35, 12, 13,
          11, 34, 39,  1, 10, 28, 14,  0, 24, 20],
         [45, 43, 44, 42, 41,  2,  1,  0,  3,  5,  4,  6,  8,  7, 12, 10,  9,
          11, 13, 14, 15, 16, 50, 49, 47, 46, 48],
         [43, 44, 42, 45, 41,  2,  5,  6,  3,  4,  9,  8,  1,  7, 11, 10,  0,
          12, 37, 38, 13, 36, 52, 46, 49, 40, 51],
         [42, 45, 43, 44, 41,  5,  4, 26,  6,  7,  8, 24, 25,  2, 23, 22, 13,
          12,  3,  1, 30, 14, 10, 11, 38, 27, 37],
         [44, 41, 42, 43, 45, 31, 32, 30, 33, 29, 34, 36, 37, 38, 35, 11, 28,
          12,  2, 13,  1, 10,  0, 39, 14, 27, 40],
         [ 2,  1,  0, 45, 44, 43, 42, 41,  3,  5, 52, 51,  6,  4, 53, 49,  8,
           7, 50, 46, 48, 47, 54,  9, 55, 40, 39],
         [44, 43, 41, 42, 45,  2,  1,  0,  7,  8,  6,  5,  4,  3, 10,  9, 11,
          12, 47, 48, 13, 50, 46, 49, 25, 24, 51],
         [44, 45, 42, 43, 41,  2,  1,  0,  5,  3,  4,  6, 11, 12, 10,  9,  8,
          13,  7, 14, 15, 50, 51, 17, 52, 18, 16],
         [45, 44, 42, 43, 41,  2,  9, 11,  1,  0, 12, 10,  8, 13,  6,  5,  7,
           3, 14, 18, 19, 17, 20,  4, 26, 15, 27],
         [41, 45, 42, 44, 43,  5,  4,  8, 12,  2,  9,  1,  3, 10, 11,  7,  6,
          13, 14, 15, 16, 22, 21, 19, 18, 20, 38],
         [44, 43, 41, 42, 45,  2,  1,  0,  3,  5,  4,  6,  7,  8,  9, 10, 11,
          12, 13, 53, 14, 52, 15, 51, 54, 50, 19],
         [41, 44, 43, 42, 45,  5,  7,  4,  6,  8, 40, 38, 37, 11, 39, 13, 12,
          14, 17,  9, 10, 18, 15, 16,  3, 20, 19],
         [42, 43, 44, 41, 45,  2,  1,  0,  3,  5,  6,  4,  9,  8, 10,  7, 11,
          12, 13, 38, 40, 39, 53, 28, 29, 27, 30],
         [44, 43, 41, 42, 45,  2,  3,  1,  4,  5,  6,  0, 52, 11, 12, 53, 13,
          51, 49, 50, 14, 10,  9, 20, 15, 21,  8],
         [45, 43, 44, 42, 41,  2,  1,  0,  3,  4,  5,  6, 14, 11, 12, 13, 15,
          46, 47, 32, 31, 52, 33, 20, 51, 18, 17],
         [42, 43, 44, 41, 45,  2,  1,  0,  5,  3,  4,  6,  9,  8,  7, 11, 12,
          10, 13, 14, 46, 36, 53, 35, 34, 47, 48],
         [45, 44, 43, 42, 41,  2,  3,  1,  0,  6,  5,  7,  8,  4, 11,  9, 12,
          10, 13, 15, 18, 19, 14, 20, 21, 22, 16],
         [44, 45, 43, 42, 41,  2,  1,  0,  8,  5,  7, 11,  4,  6, 10, 12,  9,
          50, 18, 19,  3, 13, 17, 51, 15, 16, 25],
         [45, 43, 44, 42, 41,  2,  1,  0,  3,  5,  6,  4, 11, 13,  9, 46, 12,
           8,  7, 47, 14, 10, 15, 52, 49, 37, 38],
         [45, 44, 42, 43, 41,  2,  1,  0, 11, 12, 13, 10, 31, 32, 14, 30, 33,
          29, 17, 18, 16, 19, 34, 15, 20,  9, 28],
         [44, 45, 43, 42, 41, 50, 49, 46, 47, 51, 53, 48, 52, 54, 10, 11, 55,
           9,  8,  7, 16, 17, 40, 37, 38, 15, 12],
         [45, 44, 42, 43, 41,  2,  1,  0,  3,  6,  8, 10,  9,  7, 11,  5,  4,
          12, 13, 31, 14, 32, 18, 20, 17, 19, 33]]], device='cuda:0')
tensor([[[45, 41, 43, 44, 42,  2,  1,  0,  3,  5,  6,  4,  7, 10,  8,  9, 11,
          12, 13, 18, 17, 19, 14, 16, 15, 30],
         [43, 42, 44, 41, 45,  2,  1,  0,  8,  5,  6,  9,  7, 11, 12,  4, 10,
          40, 38, 37, 39, 13,  3, 36, 46, 14],
         [ 2, 45, 42, 43, 44,  1, 41,  0, 31, 32, 30, 33, 29, 34, 37, 38, 13,
          36, 12, 11, 35, 14, 20, 39, 19,  3],
         [44, 45, 42, 43, 41,  2,  1,  0,  5,  3,  4,  6, 46, 11, 12, 48, 47,
          49, 13, 38, 10, 50, 37, 51, 40,  8],
         [44, 43, 45, 42, 41,  2,  5,  6,  4,  3, 11,  1, 13, 12,  0, 10, 14,
           9, 46, 15, 20, 17, 37, 19, 18, 36],
         [44, 45, 42, 43, 41,  2,  1,  0,  3,  5,  6,  8,  7,  4,  9, 46, 50,
          51, 11, 10, 52, 53, 48, 47, 12, 54],
         [45, 41, 43, 44, 42,  2,  1,  0,  3,  5,  4,  6, 11, 12, 10, 13,  7,
          17, 18, 15,  8,  9, 19, 14, 16, 24],
         [43, 44, 45, 42, 41,  2,  1,  0,  3, 38, 37, 11, 10, 39, 36, 12, 31,
          32,  8,  9, 35, 30, 33, 40, 46, 34],
         [45, 41, 43, 44, 42,  2,  1,  0,  6,  5,  7,  4,  3,  8,  9, 10, 11,
          12, 13, 14, 15, 23, 24, 21, 22, 25],
         [ 2,  1,  0, 45, 44, 42, 43, 41,  3, 53,  8,  5,  7, 52,  6, 54,  4,
          51, 55,  9, 46, 49, 50, 48, 47, 10],
         [44, 45, 43, 42, 41,  2,  1,  0,  3,  5,  6, 11, 12, 10,  4,  9,  8,
          36, 13, 32, 37, 38, 31, 33, 30, 46],
         [45, 41, 43, 44, 42,  5,  2,  4,  3,  6,  1,  8,  7,  9, 11, 12, 10,
          13,  0, 14, 20, 17, 19, 21, 15, 16],
         [43, 44, 45, 42, 41,  2, 32, 31, 33, 30, 36, 37, 34,  1, 38, 35, 12,
          11, 13, 39, 29, 14,  5,  4,  3, 20],
         [45, 41, 43, 44, 42,  2,  1,  0,  5,  4,  6,  3, 11, 30, 12, 31, 13,
          10, 32, 19, 33, 17, 18,  9,  7, 16],
         [41, 42, 44, 45, 43,  5,  9,  8,  6,  7,  2, 10, 11,  4,  1, 12, 48,
          25, 24, 22, 23, 46, 50,  3, 47, 21],
         [43, 42, 44, 45, 41,  5,  2,  4,  3,  1, 46,  6, 53,  8, 47, 49, 11,
          51, 48,  7, 14, 54, 52, 38, 50, 12],
         [45, 41, 43, 44, 42,  2,  1,  0,  3,  5,  4,  6, 30, 12, 11, 34, 32,
          36, 29, 31, 35, 10, 33, 19, 37, 13],
         [42, 41, 43, 44, 45,  2,  1,  0, 31, 32, 30, 33, 29, 34, 35, 36, 28,
          37, 38, 39, 11, 13, 12, 17, 19, 18],
         [45, 44, 43, 41, 42,  2,  1,  0,  6,  5,  8,  7,  9,  4, 10, 11, 12,
          17, 13,  3, 18, 14, 19, 20, 16, 31],
         [45, 41, 43, 44, 42,  2,  1,  0,  3,  5,  6,  4, 11,  8, 12,  7, 10,
          13,  9, 36, 32, 31, 30, 37, 38, 33],
         [43, 42, 44, 41, 45,  2,  1,  0,  3,  5,  8,  4,  6,  7, 10,  9, 11,
          18, 17, 19, 16, 15, 12, 50, 25, 24],
         [45, 41, 43, 44, 42,  2,  1,  0,  5, 31, 30, 32,  3,  6, 33,  4, 34,
          29,  7,  8, 11, 12, 35,  9, 10, 13],
         [45, 44, 42, 43, 41,  2,  1,  0,  3,  5,  4,  6,  8,  7,  9, 10, 11,
          50, 12, 53, 51, 54, 52, 13, 15, 16],
         [45, 41, 43, 44, 42,  2,  1,  0,  3,  5,  6,  4,  8, 12, 11,  7, 13,
           9, 10, 14, 15, 18, 17, 36, 37, 19],
         [45, 44, 43, 42, 41,  5,  6,  4,  2,  7,  8,  3,  1, 46,  9, 11, 12,
          10, 13,  0, 48, 49, 53, 47, 51, 54],
         [43, 42, 44, 41, 45,  2,  1,  0,  5,  3,  4,  6,  8,  7, 12, 11,  9,
          10, 46, 13, 14, 16, 15, 17, 18, 47],
         [ 2,  1,  0, 43, 45, 44, 42, 41,  5,  6, 52,  3,  4, 12, 11, 46, 53,
          13, 20, 19, 54, 55, 37, 38, 49, 14],
         [45, 42, 43, 44, 41,  2,  1,  0, 11, 12, 13,  5,  6, 10, 14,  4,  3,
           9, 17, 18, 19, 20,  8, 16, 15, 37],
         [44, 43, 41, 42, 45,  2,  1,  0,  3,  5,  4,  6,  8,  7, 46, 11,  9,
          12, 10, 38, 47, 50, 48, 39, 37, 13],
         [45, 41, 43, 44, 42,  2,  3,  5,  1,  4,  6,  0,  9,  8, 10,  7, 11,
          12, 13, 14, 15, 38, 16, 17, 40, 21],
         [45, 41, 43, 44, 42,  2,  1,  0,  3,  5,  4,  6, 11, 12, 13, 10,  9,
           8, 14,  7, 15, 17, 18, 19, 20, 16],
         [42, 43, 41, 44, 45, 37, 31, 32, 38, 36, 30, 34, 33, 35,  2,  5,  6,
          29, 11, 12, 39,  3,  4, 10, 13, 40]]], device='cuda:0')
tensor([[[13,  2, 14, 17, 16, 15, 12, 11,  3,  1, 31, 32, 30,  0,  5,  6, 33,
          10, 49, 50,  4, 18, 29, 19, 52],
         [ 2,  1,  0,  3,  5,  6,  4, 11, 10,  8,  9, 12,  7, 13, 14, 15, 42,
          45, 17, 43, 44, 16, 18, 41, 19],
         [11, 12, 46, 43, 13, 44, 45,  5,  9, 10, 42,  6, 49,  4, 14, 47, 38,
          37,  2, 40, 48,  3, 51, 36,  8],
         [ 2,  1,  0,  3,  5,  4,  6, 11, 12, 46, 31, 45, 44, 30, 10, 32,  8,
          43, 38, 13, 37, 36, 29,  7, 33],
         [46, 45, 38, 37, 43, 36, 44, 42, 39, 32, 30, 31, 33, 40, 49, 48, 47,
          35, 34, 29, 41, 12, 11, 50, 52],
         [30, 12, 31, 11, 32, 13, 36, 38, 33, 37, 35, 29, 34,  5, 10, 14,  2,
           4, 25, 39, 24,  3, 23, 28, 26],
         [ 5, 11,  9,  2, 12, 13,  8, 10,  4,  6,  3,  7, 14,  1, 46, 50, 49,
          48, 47, 44, 43, 45, 15, 36, 51],
         [ 2,  5,  3,  4,  1,  6, 11, 12,  0,  9, 13, 10,  8, 14,  7, 19, 15,
          20, 18, 17, 16, 21, 22, 30, 29],
         [14, 15, 17, 16,  4,  3,  2, 20, 23, 22,  5, 21,  6, 18, 24, 13, 31,
          32, 33, 34, 35, 52, 53, 30, 19],
         [ 2,  1,  0,  3,  5,  4,  7,  6,  8,  9, 10, 11, 12, 13, 14, 15, 16,
          18, 19, 20, 22, 21, 17, 46, 26],
         [ 5,  2,  4,  3,  1,  6,  8,  7,  9,  0, 11, 12, 10, 13, 14, 15, 53,
          54, 16, 17, 52, 18, 51, 21, 50],
         [30, 31, 32, 33, 19, 18, 20, 34, 37, 17, 36, 13, 11, 38, 12, 29, 35,
           5, 16,  2, 14,  4, 39, 46,  3],
         [ 2,  5,  6,  4,  3,  1,  7,  8,  9, 11, 12, 13,  0, 10, 14, 15, 17,
          16, 20, 19, 18, 21, 22, 23, 32],
         [ 2,  1,  0, 31, 32, 30, 33, 11, 13, 12, 14,  3, 29, 17, 20, 18, 15,
          19, 16, 34, 28, 10, 24, 36, 35],
         [11, 13, 12,  5, 14, 49, 50, 51, 10,  4, 18, 52, 53, 19, 15, 17,  2,
           3, 20, 44, 16,  9, 46, 43,  1],
         [ 5,  4,  2,  3,  6,  7,  1,  8,  0,  9, 10, 11, 12, 15, 14, 16, 13,
          17, 22, 21, 18, 23, 53, 24, 52],
         [11,  8, 10,  9, 12, 48, 46, 47,  7, 50, 49, 51,  6, 53, 45, 13, 54,
           5, 52, 44,  4, 18, 19, 17, 14],
         [ 2,  1,  0,  3, 12, 11, 13,  5, 14, 10,  8,  9,  4,  7,  6, 15, 16,
          18, 17, 20, 19, 30, 31, 29, 32],
         [ 5,  2,  4,  3,  6, 11, 12, 19, 20, 13, 10, 21,  9, 22, 23, 14, 38,
          37, 36, 46, 18,  8, 39, 33, 32],
         [ 2, 11, 12,  1, 13, 14,  5,  0,  3, 10,  9,  4, 15,  6,  8,  7, 16,
          17, 20, 18, 21, 22, 19, 23, 24],
         [ 2,  1,  0,  3,  5,  4,  6,  8, 11, 12,  7, 10,  9, 13, 14, 15, 16,
          17, 18, 19, 20, 21, 22, 23, 24],
         [11,  5, 12, 13,  2,  4,  6,  3,  9, 10, 14, 32, 31, 20, 33,  8, 30,
           7,  1, 18, 15, 34, 19, 35, 21],
         [11, 13, 12, 14, 15, 10,  2,  1,  0,  3,  7, 45, 43, 44, 47,  6, 46,
           4,  5, 18, 17, 16, 20, 19,  9],
         [ 2,  1,  0,  3, 11, 12, 13, 10, 14,  5, 17, 18, 19, 16,  6, 31, 15,
           4,  9, 32, 30, 33, 20,  8, 29],
         [46, 44, 45, 43, 42, 38, 37, 39, 52, 36, 40, 53, 32, 35, 31, 33, 30,
          18, 54, 19, 12, 34, 47, 11, 13],
         [10,  9,  8, 11, 12,  7,  6, 13, 50, 51,  5, 52, 53, 48, 46, 49,  4,
          14, 54, 47, 45, 38, 42, 44, 37],
         [ 2,  3,  5,  4,  1,  6,  0, 46, 13, 11, 45, 44, 12, 47, 43, 14, 50,
          48, 10, 17, 52, 51, 49, 31, 15],
         [32, 31, 30, 33, 29, 34, 28, 13, 12, 11, 14, 35, 36, 10, 37, 38, 15,
          20, 39, 17, 52, 53, 21, 23,  5],
         [ 2,  1,  0,  3,  5,  4,  6,  8, 12,  9,  7, 11, 14, 13, 10, 15, 20,
          23, 22, 16, 17, 21, 24, 18, 31],
         [ 5, 11, 12, 46,  9,  8,  4, 13, 49,  6, 43, 45, 10, 44, 42, 48,  7,
          52, 50, 51,  2, 40, 47, 38,  3],
         [32, 31, 33, 30, 29, 20, 36, 19, 37, 18,  5, 38, 34, 35, 28, 17,  4,
          26, 27, 39, 24, 25,  6, 16, 11],
         [12, 13, 11,  2, 14, 31,  1, 30, 32,  0, 10, 33, 29, 17, 18, 20, 15,
          34,  3, 16, 19, 28,  9, 36, 35]]], device='cuda:0')