Pytorch中的dataset类——创建适应任意模型的数据集接口