to_graph

torchdata.datapipes.utils.to_graph(dp, *, debug: bool = False) → graphviz.Digraph

通过返回一个 graphviz.Digraph 来可视化数据管道，这是一个数据管道的图形。这允许您直观地检查在数据管道中发生的所有转换。

注意

需要使用 graphviz 包来使用此函数。

注意

返回的图形对象的常用接口为

• render(): 将图形保存到文件。

• view(): 在查看器中打开图形。

参数：:

• dp – 您想要可视化的数据管道（通常是数据管道链中的最后一个）。

• debug (bool) – 如果为 True，则呈现通常对用户隐藏的内部数据管道（例如 ChildDataPipe，即 demux 和 fork 的子级）。默认值为 False。

示例

>>> from torchdata.datapipes.iter import IterableWrapper
>>> from torchdata.datapipes.utils import to_graph
>>> dp = IterableWrapper(range(10))
>>> dp1, dp2 = dp.demux(num_instances=2, classifier_fn=lambda x: x % 2)
>>> dp1 = dp1.map(lambda x: x + 1)
>>> dp2 = dp2.filter(lambda _: True)
>>> dp3 = dp1.zip(dp2).map(lambda t: t[0] + t[1])
>>> g = to_graph(dp3)
>>> g.view()  # This will open the graph in a viewer