Would a simple dict work?
idx_to_label = {
    0: 'class0',
    1: 'class1',
    2: 'class2'
}

preds = torch.argmax(torch.randn(10, 2), 1)
for pred in preds:
    print(idx_to_label[pred.item()])

or what do you want to do with these labels?