torchwrench.utils.data.slicer module¶

class torchwrench.utils.data.slicer.DatasetSlicer(*, add_slice_support: bool = True, add_indices_support: bool = True, add_mask_support: bool = True, add_none_support: bool = True)[source]¶

Bases: Generic[T], ABC, Dataset[T]

abstractmethod get_item(idx, /, *args, **kwargs) → Any[source]¶

get_items_indices(indices: Iterable[int] | Tensor | ndarray, *args) → list[T][source]¶

get_items_mask(mask: Iterable[bool] | Tensor | ndarray, *args) → list[T][source]¶

get_items_none(none: None, *args) → list[T][source]¶

get_items_slice(slice_: slice, *args) → list[T][source]¶

class torchwrench.utils.data.slicer.DatasetSlicerWrapper(dataset: SupportsGetitemLen[T, Any], *, add_slice_support: bool = True, add_indices_support: bool = True, add_mask_support: bool = True, add_none_support: bool = True)[source]¶

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

get_item(idx: int, *args) → T[source]¶