A multi-class classification would usually contain a single class index for each sample.
For the posted input shape with a batch size of 1, your target would have the shape [batch_size=1] and contain a value in [0, nb_classes-1] assuming you are using nn.CrossentropyLoss.