Shortcuts

list_dps

torchdata.dataloader2.graph.list_dps(graph: Dict[int, Tuple[Union[IterDataPipe, MapDataPipe], Dict[int, Tuple[Union[IterDataPipe, MapDataPipe], DataPipeGraph]]]], exclude_dps: Optional[Union[IterDataPipe, MapDataPipe, List[Union[IterDataPipe, MapDataPipe]]]] = None) List[Union[IterDataPipe, MapDataPipe]]

Given the graph of DataPipe generated by traverse_dps function, return a list of all DataPipe instances without duplication. If exclude_dps is provided, the provided DataPipes and their predecessors will be ignored.

Note

  • The returned list is in the order of breadth first search of the graph

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources