在pytorch中dataset的格式

PyTorch中的dataset格式通常是一个包含输入数据和标签的元组。每个样本都存储在一个元组中，并作为数据集的一个元素返回。通常，输入数据是一个张量或一个numpy数组，标签是一个整数或一个张量。以下是一个示例数据集的格式：

class MyDataset(torch.utils.data.Dataset):
    def __init__(self, data, labels):
        self.data = data
        self.labels = labels
        
    def __getitem__(self, index):
        x = torch.tensor(self.data[index], dtype=torch.float32)
        y = torch.tensor(self.labels[index], dtype=torch.long)
        return x, y
    
    def __len__(self):
        return len(self.data)

在这个示例中，数据集包含data和labels两个数组。`__getitem__`方法将每个样本转换为张量，并返回一个元组(x, y)，其中x是输入数据，y是标签。`__len__`方法返回数据集的长度，即样本数。