Do I correctly understand that a shape of labels in classification and shape of output of model(input) is different?
From CIFAR-10 output = model(input), output:
tensor([[-0.3947, -1.8114,  0.5413,  1.2491, -0.0409,  2.1655, -0.7398,  0.8994,
         -0.7245, -0.6783],
        [ 5.3390,  3.6180, -0.4063, -2.5301, -2.6485, -3.5857, -3.7007, -3.5152,
          9.2215,  3.2827],
        [-1.0966, -3.1175,  1.1547,  2.9437,  0.3493,  3.4713,  0.2951,  1.2098,
         -1.7341, -2.9322],
        [-1.8294, -2.7313,  0.9235,  0.6192,  3.2165,  1.3419,  1.4856,  2.6439,
         -3.9445, -2.5060]], grad_fn=<ThAddmmBackward>)

Labels:
tensor([7, 3, 9, 8])

It’s contrintuitive to have.