feed_data

产品支持情况

产品 是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas A2 推理系列产品

函数功能

将数据输入到Graph。

函数原型

feed_data(feed_dict, flow_info=None, timeout=-1, partial_inputs = False) -> int

参数说明

参数名称 数据类型 取值说明
feed_dict Dict[FlowData, Union["numpy.ndarray", Tensor, List]] key为FlowData节点,value是可转换成numpy.ndarray的任意输入,或者dataflow.tensor。
当feed_dict为空时,flow_info必须的flow_flags必须包含DATA_FLOW_FLAG_EOS,此时partial_inputs不起作用。
flow_info FlowInfo 按需设置FlowInfo,具体请参见dataflow.FlowInfo
timeout int 数据输入超时时间,单位:ms,取值范围[0, 2147483647), 取值为-1时表示从不超时。
partial_inputs bool 每次调用feed_data接口时,feed_dict是否支持模型的部分输入,取值如下。

- True:feed_dict中可以只包含模型的部分输入。
- False:feed_dict中必须包含模型所有的输入。

默认为False。

返回值

正常场景下返回0。

异常场景下返回具体的错误码,并打印错误日志。

调用示例

import dataflow as df
graph = df.FlowGraph(...)
graph.feed_data(...)

约束说明