Mean of each class of a dataset

[S_M]

How can I compute the mean of each class in a dataset using Pytorch?