torchoutil.utils.data.dataset module

class EmptyDataset(
*args,
**kwds,
)[source]

Bases: Dataset[None]

Dataset placeholder. Raises StopIteration if __getitem__ is called.

class IterableSubset(
dataset: SupportsGetitemLen[T],
indices: Iterable[int] | LongTensor1D,
)[source]

Bases: IterableWrapper[T], Generic[T]

class IterableTransformWrapper(
dataset: SupportsGetitemLen[T],
transform: Callable[[T], U] | None,
condition: Callable[[T, int], bool] | None = None,
)[source]

Bases: IterableWrapper[T], Generic[T, U]

property condition: Callable[[T, int], bool] | None
property transform: Callable[[T], U] | None
class IterableWrapper(
dataset: SupportsGetitemLen[T],
)[source]

Bases: Generic[T], IterableDataset[T], Wrapper[T]

class Subset(
dataset: SupportsGetitemLen[T],
indices: Iterable[int],
)[source]

Bases: Generic[T], Subset[T], Wrapper[T]

dataset: Dataset[T_co]
indices: Sequence[int]
class TransformWrapper(
dataset: SupportsGetitemLen[T],
transform: Callable[[T], U] | None,
condition: Callable[[T, int], bool] | None = None,
)[source]

Bases: Generic[T, U], Wrapper[T]

property condition: Callable[[T, int], bool] | None
property transform: Callable[[T], U] | None
class Wrapper(
dataset: SupportsGetitemLen[T],
)[source]

Bases: Generic[T], Dataset[T]

unwrap(
recursive: bool = True,
) → SupportsGetitemLen | Dataset[source]