Your current label and output shape do not match. Both should have the batch size in dim0.
Based on your output shape it looks like you are dealing with a single sample and 16 classes. If that’s correct, your label should have the shape [1] and contain the class index in the range [0, 15].