I have prototypes and each class-label.
I want to get centroid of each class. How to compute centroid ?
Here is the shape of prototypes.
print(prototypes.shape): [64,31] # batch_size : 64, number of class: 30