DataLoader 接口

DataLoader接口是一个在PyTorch中用于加载和预处理数据的工具。它提供了一种方便的方式来迭代和批量处理数据，以供模型训练和评估使用。

DataLoader接口的主要功能包括：
1. 数据加载：可以从多种数据源中加载数据，如内存中的张量、文件系统中的图像或文本数据等。
2. 数据预处理：可以对加载的数据进行各种预处理操作，如图像的裁剪、缩放、标准化等。
3. 数据批处理：可以将数据按照指定的批次大小进行分组，方便模型进行批量计算。
4. 数据迭代：可以通过迭代器的方式逐批次地提供数据，方便模型进行训练和评估。

使用DataLoader接口可以大大简化数据处理的流程，提高代码的可读性和可维护性。以下是使用DataLoader接口的一些示例代码：

import torch
from torch.utils.data import DataLoader, Dataset

# 自定义数据集类
class MyDataset(Dataset):
    def __init__(self, data):
        self.data = data

    def __getitem__(self, index):
        return self.data[index]

    def __len__(self):
        return len(self.data)

# 创建数据集实例
dataset = MyDataset([1, 2, 3, 4, 5])

# 创建DataLoader实例
dataloader = DataLoader(dataset, batch_size=2, shuffle=True)

# 迭代数据
for batch in dataloader:
    print(batch)

上述代码中，首先定义了一个自定义的数据集类`MyDataset`，然后创建了一个数据集实例`dataset`，并将其传入DataLoader中。通过设置`batch_size`参数为2，表示每次迭代返回2个样本。在迭代过程中，可以通过`for`循环逐批次地获取数据。