Dask可视化DASK图中的多个输出节点

Question

我正在创建的DASK图有多个输出。

我想知道是否可以同时可视化多个DASK输出。

尝试使用dask.visualize(graph)时。其中 graph是dask节点的元组或字典。它似乎因产生空图像而失败。

例如：

import dask
@dask.delayed
def op(x, y):
    return x + y
def graph():
    a = 1
    b = 2
    c = 3
    d = 4
    f = op(a, b)
    g = op(c, d)
    h = op(a, d)
    output1 = op(f, g)
    output2 = op(g, h)
    # works
    dask.visualize(output1, filename='output1.png')
    dask.visualize(output2, filename='output2.png')
    # empty small square box
    dask.visualize((output1, output2), filename='output1and2.png')
    # only outputs the name of the keys
    dask.visualize({'output1': output1, 'output2': output2}, filename='output1and2_.png')
graph()

Answer 1

dask.visualize函数需要许多参数，而不是一系列参数

做这个

dask.visualize(output1, output2, filename='output1and2.png')

而不是这个

dask.visualize((output1, output2), filename='output1and2.png')