pytorch中使用共享内存缓存数据集，避免发生复制