昇腾社区首页
中文
注册

运行FlowGraph

功能介绍

构建完FlowGraph之后,如果您希望直接运行FlowGraph,得到图的执行结果,可以参考本节内容。涉及的主要接口为:

  1. 调用feed_data,设置Graph输入。
  2. 调用fetch_data,获取Graph输出。
  3. 调用dataflow.finalize,释放系统资源。

使用DataFlow开发框架时,NN模型执行使用是饱和模式。

开发示例

  1. 包含的模块。
    1
    import dataflow as df
    
  2. 设置Graph输入。

    示例代码如下:

    1
    2
    3
    4
    5
    6
    7
    # feed data
    feed_data0 = np.ones([1,2], dtype=np.int32)
    feed_data1 = np.array([[1,2]], dtype=np.int32)
    flow_info = df.FlowInfo()
    flow_info.start_time = 0
    flow_info.end_time = 5
    dag.feed_data({data0:feed_data0, data1:feed_data1}, flow_info) # 异步喂数据
    
  3. 获取Graph输出。
    示例代码如下:
    1
    2
    # fetch data
    result = dag.fetch_data() # 异步取结果
    

    将Graph输出数据保存在result中。

  4. 图运行完之后,通过dataflow.finalize释放资源。
    1
    2
    # 释放dataflow资源
    df.finalize()