Thanks so much, Ptrblck for the help. I managed to do that as per your instructions, however, the displayed features appear on the corresponding class indices in a straight line (figure given below)   instead of clusters. Moreover, these features are from the 1st epoch and are generated at runtime. I have already an embedded feature file (feature.pth) for all epochs, could you please guide how to use this file to display the features in a clustered display.
feats_disp1449×732 64.7 KB

Cheers.