I want to compute the classification accuracy, while my targets and predictions have the shape (batch_size, nb_classes, x_size_image, y_size_image). The accuracy is the number of correctly classified pixels divided by the total number of pixels. Is there an easy way to do that?