torch.utils.data.ConcatDataset
时间: 2023-12-11 13:29:16 浏览: 349
合并两个DataSet的数据内容的方法
torch.utils.data.ConcatDataset是PyTorch中的一个数据集类,用于将多个数据集合并成一个大的数据集。合并后的数据集可以像普通数据集一样使用,例如可以进行数据加载、数据扩充等操作。
使用方法如下:
```python
from torch.utils.data import ConcatDataset
dataset1 = ...
dataset2 = ...
concat_dataset = ConcatDataset([dataset1, dataset2])
```
阅读全文