Python 正在加载.npy文件作为pytorch的数据集

我已经对.npy文件中的数据进行了预处理，对于原始数据我们称之为X.npy，对于标签我们称之为Y.npy。它们的组织方式是匹配两个文件中的每个元素（X中的第一个元素具有Y中的第一个标签等）。如何使用

torch.utils.data.DataLoader

将其加载为数据集？我是pytorch的新手，任何帮助都会很有用。

您也可以使用DatasetFolder，它基本上是ImageFolder的基础类。使用这个类，您可以提供自己的文件扩展名和加载程序来加载示例

def npy_加载器（路径）：
从_numpy返回火炬（np.装载（路径））