在PyTorch中,数据加载器可以通过torch.utils.data.DataLoader
来实现。数据加载器可以帮助用户批量加载数据,并可以在训练过程中对数据进行随机排列、并行加载等操作。
下面是一个简单的示例,演示如何使用数据加载器来加载一个简单的数据集:
import torch from torch.utils.data import Dataset, DataLoader # 创建一个自定义的数据集类 class CustomDataset(Dataset): def __init__(self): self.data = https://www.yisu.com/ask/torch.randn(100, 3) # 100个3维的随机数据'Batch {i}:') print('Data:', data) print('Target:', target)
在上述示例中,首先定义了一个自定义的数据集类CustomDataset
,然后创建了一个数据集实例dataset
。接着利用DataLoader
类来创建一个数据加载器实例data_loader
,并指定了批量大小为32且开启了数据随机排列。最后通过对数据加载器进行遍历,便可以逐批次地获取数据和标签。