Shortcuts

list_dps

torchdata.dataloader2.graph.list_dps(graph: Dict[int, Tuple[IterDataPipe | MapDataPipe, Dict[int, Tuple[IterDataPipe | MapDataPipe, DataPipeGraph]]]], exclude_dps: IterDataPipe | MapDataPipe | List[IterDataPipe | MapDataPipe] | None = None) List[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