使用PyTorch加载和预处理数据