DataIterator

class DataIterator

Bases: object

Base class that every data iterator inherits.

Example:

class CustomDataIterator(DataIterator):
    def __call__(self, inputs: Tensor, targets: Tensor) -> Iterator[Batch]:
        for inp, targ in zip(inputs, targets):
            yield Batch(inp, targ)

__init__()

Initialize self. See help(type(self)) for accurate signature.

Methods

__init__()

Initialize self.