remove_dp

torchdata.dataloader2.graph.remove_dp(graph: Dict[int, Tuple[Union[IterDataPipe, MapDataPipe], Dict[int, Tuple[Union[IterDataPipe, MapDataPipe], DataPipeGraph]]]], datapipe: Union[IterDataPipe, MapDataPipe]) → Dict[int, Tuple[Union[IterDataPipe, MapDataPipe], Dict[int, Tuple[Union[IterDataPipe, MapDataPipe], DataPipeGraph]]]]

给定由 traverse_dps 函数生成的 DataPipe 图和要删除的 DataPipe，返回新的 DataPipe 图。

注意

• 此函数无法删除将多个 DataPipes 作为输入的 DataPipe。