Hi, thanks for the reply. I’m still having trouble following how to understand the cross entropy loss. Judging by your answer, it seems that the C that we’re giving our loss function is the correct class?
However, in my case I still get the same runtime error if I do
loss(output, torch.tensor([3]))

where output.shape = 85997.
Also, if C = # of classes as the documentation says, shouldn’t C = torch.tensor([10]) in the example you gave?