[ 仅 API 调用方式不一致 ]torch.utils.data.Dataset

torch.utils.data.Dataset(*args, **kwargs)

paddle.io.Dataset()

# PyTorch 写法
class Data(torch.utils.data.Dataset):
    def __init__(self):
        self.x = [1, 2, 3, 4]

    def __getitem__(self, idx):
        return self.x[idx]

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

# Paddle 写法
class Data(paddle.io.Dataset):
    def __init__(self):
        self.x = [1, 2, 3, 4]

    def __getitem__(self, idx):
        return self.x[idx]

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