Как отсортировать набор данных в pytorch

Question

Я хотел бы отсортировать мой набор данных по числовым значениям в метках.

Есть ли в Pytorch функция для эффективного управления этим?

мой набор данных type() состоит из:

 <class 'torchvision.datasets.mnist.MNIST'>

Answer 1

Не существует универсального способа сделать это эффективно, поскольку класс набора данных реализует только методы __getitem__ и __len__ и не обязательно должен иметь какую-либо «хранимую» информацию о метках.

Однако в случае класса MNIST вы можете отсортировать набор данных из списка меток.

Например, если вы хотите перечислить индексы с меткой 5.

mnist = torchvision.datasets.mnist.MNIST("/")
labels = mnist.train_labels
fives = (labels == 5).nonzero()