在PyTorch数据集中使用的len函数在哪里？

import torch
from torch.utils.data import Dataset

class CustomDataset(Dataset):
    def __init__(self, data):
        self.data = data

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

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

# 创建数据集对象
data = [1, 2, 3, 4, 5]
dataset = CustomDataset(data)

# 使用len函数获取数据集大小
dataset_size = len(dataset)
print("数据集大小：", dataset_size)