Sorry, I don’t know.
I am currently studying face recognition. In face recognition, such as ArcFace, center of features of each class is a learnable matrix, which shape is (embedding_len, num_classes).