I have this problem with the dimensions, can someone explain it better?
This means that the output of your network is not of rhe same shape as your labels, the logits i.e. predictions should be the same shape as labels (32, ). Check the output of you model.