calculate Entropy for each class of the test set to measure uncertainty on pytorch

Question

I am trying to calculate Entropy of each class of the dataset for an image classification task to measure uncertainty on pytorch,using the MC Dropout method and the solution pro