find_dps¶
- torchdata.dataloader2.graph.find_dps(graph: Dict[int, Tuple[Union[IterDataPipe, MapDataPipe], Dict[int, Tuple[Union[IterDataPipe, MapDataPipe], DataPipeGraph]]]], dp_type: Type[Union[IterDataPipe, MapDataPipe]]) List[Union[IterDataPipe, MapDataPipe]] ¶
给定由
traverse_dps
函数生成的 DataPipe 图,返回具有提供的 DataPipe 类型的 DataPipe 实例。