traverse_dps¶
- torchdata.dataloader2.graph.traverse_dps(datapipe: Union[IterDataPipe, MapDataPipe]) Dict[int, Tuple[Union[IterDataPipe, MapDataPipe], Dict[int, Tuple[Union[IterDataPipe, MapDataPipe], DataPipeGraph]]]] ¶
遍历 DataPipes 及其属性以提取 DataPipe 图。
这只会查看每个 DataPipe 中的属性,这些属性是 DataPipe 以及 Python 集合对象,例如
list
、tuple
、set
和dict
。- 参数:
datapipe – 图的末尾 DataPipe
- 返回:
以嵌套字典表示的图,其中键是 DataPipe 实例的 ID,值是 DataPipe 实例和子图的元组