Based on the shape of output it looks like you are working on some segmentation task with 16 classes.
If that’s the case, your target should have the shape [10, 52, 2].
In the usual multi-class classification use case, you would provide the output as [batch_size, nb_classes] and the target as [batch_size] containing the class indices.
Have a look at the docs 30 for more shape information.