Попробуйте ввести код.
Random rnd = new Random();
int month = rnd.Next(1, 13); // creates a number between 1 and 12
int dice = rnd.Next(1, 7); // creates a number between 1 and 6
int card = rnd.Next(52); // creates a number between 0 and 51
Не существует общего способа сделать это эффективно, так как класс набора данных реализует только методы __getitem__
и __len__
и не обязательно имеет какую-либо «хранимую» информацию о метках.
В случае класса набора данных MNIST , однако, вы можете отсортировать набор данных из списка меток.
Например, когда вы хотите перечислить индексы, которые имеют метку 5.
mnist = torchvision.datasets.mnist.MNIST("/")
labels = mnist.train_labels
fives = (labels == 5).nonzero()