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

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

paddle.io.IterableDataset()

# PyTorch 写法
class MyIterableDataset(torch.utils.data.IterableDataset):
    def __init__(self, start, end):
        super(MyIterableDataset).__init__()
        self.start = start
        self.end = end

    def __iter__(self):
        return iter(range(self.start, self.end))

# Paddle 写法
class MyIterableDataset(paddle.io.IterableDataset):
    def __init__(self, start, end):
        super(MyIterableDataset).__init__()
        self.start = start
        self.end = end

    def __iter__(self):
        return iter(range(self.start, self.end))