利用torch.utils.data.Dataset自定义数据加载类