下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

feed_data

函数功能

将数据输入到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,或者python的List,或者Tensor。

当feed_dict为空时,flow_info必须的flow_flags必须包含DATA_FLOW_FLAG_EOS,此时partial_inputs不起作用

flow_info

FlowInfo

按需设置FlowInfo,具体请参见dataflow.FlowInfo

timeout

int

数据输入超时时间,单位:ms,取值为-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(...)

约束说明

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词