torchoutil.utils.data.slicer module¶

class 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]

abstract 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 DatasetSlicerWrapper( dataset: SupportsGetitemLen[T], *, add_slice_support: bool = True, add_indices_support: bool = True, add_mask_support: bool = True, )[source]¶

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

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