6.1.1 使用torch.utils.data.Dataset封装自定义数据集