MERAH_Samia:

torch.cat(output_list, 0)


Here you’re creating a sampling distribution, and only using it for .mean(0), you can use it to calculate uncertainty estimates that you need.
Something similar to  torch.cat(output_list, 0).softmax(1).sd(0) (and gather predicted classes)