Okay, just so I understand that right, you can’t pass PyTorch say [1,0] for a cat label and [0,1] for a dog label for all the training images?
Not sure I understand how you specify the dimensions [batch_size, number_of_classes] during the model building when I’ve already specified the number of outputs as two in this case  but I think I get that what you’re saying is your output will be say 10 by 2 if you have 10 training images in a batch, but when you test a single image it will come out at 1 by 2?
Thanks by the way this is really helpful