如何获取MNIST数据集对应索引值的数据形成一个新的数据集

我是采用的torchvision的datasets读取的MNIST数据集，得到的应该是个元组，train数据集里面有6000张图片，想问一下如果我想得到如[1,5,9,48,59,115]索引值的数据形成一个新的数据集，应该怎么做呢？

torch.index_select()从tensor中按维度取出指定索引的元素_anshiquanshu的专栏-CSDN博客 此方法的作用是，根据dim取出input中的index对应的元素，并且返回一个tensor，首先是参数介绍input(Tensor)：输入的tensor。 dim(int)：指定我们要进行索引选择的维度。 index(LongTensor)：索引。 out：输出tensor。使用方法：>>> x = torch.randn(3, 4)>>> xtensor([[ 0.1427, 0.0231, -0.5414, -1.0009], https://blog.csdn.net/anshiquanshu/article/details/112510796