Thanks for the prompt feedback! Yes, its a multi-class classification problem.
With regards to the points you have raised, am I right to interpret that I should transform the target (which is of current shape [32] to the shape [32, 3]? Which means transforming
tensor([1, 2, 2, 1, 1, 2, 2, 1, 2, 2, 2, 2, 2, 0, 2, 1, 0, 2, 2, 0, 2, 0, 2, 2,         1, 0, 2, 0, 0, 2, 1, 1], device='cuda:0')
to something like
tensor([[0, 1, 0], [0, 0, 1], ..., [0, 1, 0]], device='cuda:0')
for the loss function to work?