ptrblck:

target.view(target.size(0), -1).unique(dim=1)


labels.view(labels.size(0), -1).unique(dim=1)
tensor([[ 2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,
2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,
2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  2,  3,  3,
3,  3,  3,  3,  3,  3,  3,  3,  3,  3,  3,  3,  3,  3,  3,  3,  3,  3,
3,  3,  3,  3,  3,  3,  3,  3,  3,  3,  3,  5,  5,  5,  5,  5,  5,  5,
5,  5,  5,  5,  5,  5,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,  6,
6,  7,  7,  7,  7,  7,  7,  7,  7,  7,  7,  7,  7,  7,  8,  8,  8,  8,
8,  8,  8,  8,  8,  8,  8,  8,  8,  8,  8,  8,  8,  8,  8,  8,  8,  8,
8,  9,  9,  9,  9,  9,  9,  9,  9],
[ 2,  2,  2,  2,  2,  2,  2,  2,  3,  3,  3,  3,  3,  3,  4,  4,  4,  5,
5,  5,  5,  5,  6,  6,  6,  7,  7,  7,  7,  7,  8,  8,  8,  8,  8,  9,
9,  9,  9,  9,  9,  9,  9, 10, 10, 10, 10, 11, 11, 11, 11, 11,  2,  2,
2,  2,  2,  2,  2,  3,  3,  3,  3,  3,  3,  4,  6,  6,  7,  7,  7,  7,
8,  8,  8,  8,  9,  9,  9,  9,  9, 11, 11,  2,  2,  2,  3,  3,  3,  9,
9,  9, 10, 10, 11, 11,  2,  2,  2,  5,  6,  7,  7,  7,  9,  9, 10, 11,
11,  2,  2,  4,  4,  7,  7,  9,  9, 10, 10, 10, 11, 11,  2,  2,  2,  2,
2,  2,  3,  3,  3,  3,  3,  3,  6,  7,  7,  8,  8,  8,  8,  9,  9,  9,
9,  2,  2,  3,  3,  7,  7, 10, 10],
[ 1,  2,  3,  5,  6,  7,  8,  9,  2,  3,  6,  7,  8,  9,  2,  7,  9,  1,
2,  3,  8,  9,  1,  2,  6,  2,  5,  6,  7,  9,  2,  3,  6,  8,  9,  1,
2,  3,  5,  6,  7,  8,  9,  1,  2,  3,  5,  2,  3,  5,  6,  7,  2,  3,
5,  6,  7,  8,  9,  2,  3,  6,  7,  8,  9,  7,  2,  6,  2,  5,  6,  7,
2,  3,  6,  9,  2,  3,  5,  6,  7,  3,  7,  2,  3,  8,  2,  3,  8,  2,
3,  8,  2,  3,  2,  3,  2,  5,  9,  2,  2,  2,  5,  9,  2,  5,  2,  2,
5,  2,  3,  2,  3,  2,  5,  2,  3,  2,  3,  5,  2,  3,  2,  3,  6,  7,
8,  9,  2,  3,  6,  7,  8,  9,  2,  2,  6,  2,  3,  6,  9,  2,  3,  6,
7,  2,  9,  2,  9,  2,  9,  2,  3]])
I definitely sense some problem here