Hi,
Yes sure.
My input for to the data loader is as follows for 2 rows-
images–
[[tensor([[  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,  51, 159, 253, 159,  50,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
          48, 238, 252, 252, 252, 237,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  54,
         227, 253, 252, 239, 233, 252,  57,   6,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  10,  60, 224,
         252, 253, 252, 202,  84, 252, 253, 122,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0, 163, 252, 252,
         252, 253, 252, 252,  96, 189, 253, 167,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  51, 238, 253, 253,
         190, 114, 253, 228,  47,  79, 255, 168,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,  48, 238, 252, 252, 179,
          12,  75, 121,  21,   0,   0, 253, 243,  50,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,  38, 165, 253, 233, 208,  84,
           0,   0,   0,   0,   0,   0, 253, 252, 165,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   7, 178, 252, 240,  71,  19,  28,
           0,   0,   0,   0,   0,   0, 253, 252, 195,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,  57, 252, 252,  63,   0,   0,   0,
           0,   0,   0,   0,   0,   0, 253, 252, 195,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0, 198, 253, 190,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0, 255, 253, 196,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,  76, 246, 252, 112,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0, 253, 252, 148,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,  85, 252, 230,  25,   0,   0,   0,   0,
           0,   0,   0,   0,   7, 135, 253, 186,  12,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,  85, 252, 223,   0,   0,   0,   0,   0,
           0,   0,   0,   7, 131, 252, 225,  71,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,  85, 252, 145,   0,   0,   0,   0,   0,
           0,   0,  48, 165, 252, 173,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,  86, 253, 225,   0,   0,   0,   0,   0,
           0, 114, 238, 253, 162,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,  85, 252, 249, 146,  48,  29,  85, 178,
         225, 253, 223, 167,  56,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,  85, 252, 252, 252, 229, 215, 252, 252,
         252, 196, 130,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,  28, 199, 252, 252, 253, 252, 252, 233,
         145,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,  25, 128, 252, 253, 252, 141,  37,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0]],
       dtype=torch.uint8), tensor([[  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0, 187,
         254, 105,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   5, 147, 254,
         253, 243, 230,  51,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  74, 253, 254,
         253, 253, 253, 199,  53,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  34, 240, 253, 228,
          94, 202, 185, 254, 198,  17,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  47, 254, 254,  34,
           0,   0,   0,  85, 248, 163,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0, 181, 253, 202,   0,
           0,   0,   0,   0, 163, 251,  80,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,  57, 249, 253, 119,   0,
           0,   0,   0,   0,   9, 173, 202,   9,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0, 136, 253, 236,  19,   0,
           0,   0,   0,   0,   0,  49, 253, 127,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,  60, 254, 254, 171,   0,   0,
           0,   0,   0,   0,   0,  17, 235, 204,   7,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,  93, 254, 227,  29,   0,   0,
           0,   0,   0,   0,   0,   0, 167, 254,  23,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0, 168, 254, 198,   0,   0,   0,
           0,   0,   0,   0,   0,   0,  93, 254,  99,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,  17, 234, 254, 115,   0,   0,   0,
           0,   0,   0,   0,   0,   0,  93, 254, 115,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,  24, 254, 254, 115,   0,   0,   0,
           0,   0,   0,   0,   0,   0, 119, 254,  56,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,  74, 253, 254, 115,   0,   0,   0,
           0,   0,   0,   0,   0,  38, 247, 190,   2,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,  40, 253, 254, 115,   0,   0,   0,
           0,   0,   0,   0,  13, 161, 253, 101,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,  24, 253, 254, 148,   0,   0,   0,
           0,   0,   0,  26, 180, 253, 185,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,  17, 235, 255, 237,  80,   0,   0,
           0,   0, 151, 254, 254, 222,  84,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,  84, 254, 253, 251, 230, 231,
         214, 214, 247, 254, 210,  63,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0, 129, 242, 253, 253, 254,
         253, 253, 202,  80,  21,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  55, 169, 253, 254,
         202, 152,  19,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0]],
       dtype=torch.uint8), tensor([[  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0, 101, 191, 113, 113,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0, 101, 249, 252, 252, 252,
         126,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,  25,  55,  87, 233, 252,
         140,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  72, 239,
         140,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  63,
         140,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  13, 113,
         141,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  29, 252,
         140,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0, 138, 252,
         140,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0, 169, 252,
         153,   3,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0, 169, 252,
         253,  27,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0, 108, 253,
         255,  27,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  29, 252,
         253,  27,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  29, 252,
         253, 136,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  29, 252,
         253, 167,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  13, 189,
         253, 214,  31,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  79,
         255, 253, 119,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
         206, 252, 246, 122,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
          88, 246, 252, 223,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0, 122, 246, 223,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0, 119, 223,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0]],
       dtype=torch.uint8)], [tensor([[  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,  51, 159, 253, 159,  50,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
          48, 238, 252, 252, 252, 237,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  54,
         227, 253, 252, 239, 233, 252,  57,   6,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  10,  60, 224,
         252, 253, 252, 202,  84, 252, 253, 122,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0, 163, 252, 252,
         252, 253, 252, 252,  96, 189, 253, 167,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  51, 238, 253, 253,
         190, 114, 253, 228,  47,  79, 255, 168,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,  48, 238, 252, 252, 179,
          12,  75, 121,  21,   0,   0, 253, 243,  50,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,  38, 165, 253, 233, 208,  84,
           0,   0,   0,   0,   0,   0, 253, 252, 165,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   7, 178, 252, 240,  71,  19,  28,
           0,   0,   0,   0,   0,   0, 253, 252, 195,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,  57, 252, 252,  63,   0,   0,   0,
           0,   0,   0,   0,   0,   0, 253, 252, 195,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0, 198, 253, 190,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0, 255, 253, 196,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,  76, 246, 252, 112,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0, 253, 252, 148,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,  85, 252, 230,  25,   0,   0,   0,   0,
           0,   0,   0,   0,   7, 135, 253, 186,  12,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,  85, 252, 223,   0,   0,   0,   0,   0,
           0,   0,   0,   7, 131, 252, 225,  71,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,  85, 252, 145,   0,   0,   0,   0,   0,
           0,   0,  48, 165, 252, 173,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,  86, 253, 225,   0,   0,   0,   0,   0,
           0, 114, 238, 253, 162,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,  85, 252, 249, 146,  48,  29,  85, 178,
         225, 253, 223, 167,  56,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,  85, 252, 252, 252, 229, 215, 252, 252,
         252, 196, 130,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,  28, 199, 252, 252, 253, 252, 252, 233,
         145,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,  25, 128, 252, 253, 252, 141,  37,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0]],
       dtype=torch.uint8), tensor([[  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  11,  92,
         214, 253, 254, 253, 152,  71,  41,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,  41, 102, 183, 213, 252,
         253, 252, 253, 252, 253, 252, 243, 162,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,  41, 132, 253, 254, 253, 203, 122,
           0,   0,   0,   0, 102, 183, 254, 253, 123,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,  62, 223, 253, 252, 151,  70,   0,   0,
           0,   0,   0,   0,   0,   0, 172, 252, 203,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,  82, 254, 253, 224,  81,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,  21, 223, 234,  30,   0,   0,   0,   0],
        [  0,   0,   0,   0,  82, 243, 253, 171,  61,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0, 122, 253,  91,   0,   0,   0,   0],
        [  0,   0,   0,   0, 173, 253, 203,  20,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0, 102, 254, 213,   0,   0,   0,   0],
        [  0,   0,   0,   0, 253, 252, 102,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0, 102, 253, 252,   0,   0,   0,   0],
        [  0,   0,   0,   0, 254, 253, 102,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,  41, 254, 253,   0,   0,   0,   0],
        [  0,   0,   0,   0, 172, 252, 183,   0,   0,   0,  21, 102,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0, 253, 252,   0,   0,   0,   0],
        [  0,   0,   0,   0,  92, 253, 254,  91,   0,   0,  51, 253, 214,  10,
           0,   0,   0,   0,   0,   0,   0, 102, 254, 253,   0,   0,   0,   0],
        [  0,   0,   0,   0,  31, 232, 253, 252, 142,  61,  31, 232, 253, 131,
           0,   0,   0,   0,   0,   0,   0, 102, 253, 252,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,  41, 214, 253, 254, 253, 234, 233, 254, 253,
         254, 172,  41,   0,   0,   0,  31, 233, 254, 192,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,  10, 131, 213, 252, 253, 252, 253, 252,
         253, 252, 243, 203, 102, 102, 213, 252, 233,  50,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,  82, 123, 122, 203, 203,
         203, 203, 254, 253, 254, 253, 255, 233, 123,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,  50,  91,  50, 131,  91,  30,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0]],
       dtype=torch.uint8), tensor([[  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0, 108, 105,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
          57, 182,  43,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
         231, 253, 104,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
         161, 232,  55,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  54, 154,
         251, 208,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0, 162, 253,
         242, 183,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  31, 253,
         163,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  84, 253,
         163,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0, 179, 253,
         136,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0, 179, 253,
          14,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  32, 211, 255,
          14,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0, 111, 253, 253,
          14,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   8, 226, 253, 165,
           5,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0, 121, 253, 253, 119,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0, 120, 253, 226,  14,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  49, 236, 134,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0, 121, 253, 239,  64,
           0,   0,   0,   0,   0,   0,  64,  96,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   7, 226, 253, 246,
         151, 151, 239, 241, 239, 239, 246, 249, 158,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,  97, 240, 253,
         253, 253, 253, 236, 219, 224, 208,  76,  17,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0, 162, 104,
         192, 218, 253,  62,  25,  37,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0],
        [  0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0]],
       dtype=torch.uint8)]]

and the labels are–
[[1, 0], [1, 0]]

The labels have been converted into a tensor later on.