torch.utils.data.Dataset类原代码

由于`torch.utils.data.Dataset`类是PyTorch中的一个抽象类，因此没有原始代码。但是，我们可以查看其文档和源代码来了解其属性和方法。

`torch.utils.data.Dataset`类是一个抽象类，用于表示数据集。它有两个必须实现的方法：`__len__`和`__getitem__`。`__len__`方法返回数据集的大小，`__getitem__`方法返回给定索引的数据和标签。

以下是`torch.utils.data.Dataset`类的一些常用属性和方法：

属性：
- `dataset`: 数据集的列表。

方法：
- `__len__(self)`: 返回数据集的大小。
- `__getitem__(self, index)`: 返回给定索引的数据和标签。
- `__add__(self, other_dataset)`: 将两个数据集连接在一起。
- `__iter__(self)`: 返回一个迭代器，用于遍历数据集。
- `__next__(self)`: 返回下一个数据和标签对。
- `__repr__(self)`: 返回数据集的字符串表示形式。

以下是一个简单的示例，展示了如何创建一个自定义数据集类，继承`torch.utils.data.Dataset`类，并实现`__len__`和`__getitem__`方法：

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):
        x = self.data[index][0]
        y = self.data[index][1]
        return torch.tensor(x), torch.tensor(y)