Atlas系列硬件: Atlas 200l DK A2 开发者套件; Atlas 200l DK 开发者套件（型号：3000）; 加速模块、加速卡、智能小站、服务器、集群

异构计算架构: CANN社区版; CANN商用版

AI框架: MindSpore; TensorFlow框架适配（社区版）; TensorFlow框架适配（商用版）; PyTorch框架适配

分布式训练加速套件: MindSpeed

昇腾推理引擎: MindIE

应用使能MindX: MindX DL; MindX Edge; MindX SDK

全流程开发工具: MindStudio

评分

提单

论坛

小AI

接口列表

本文档主要描述UDF（User Define Function）模块对外提供的接口，用户可以调用这些接口进行自定义处理函数的开发，然后通过DataFlow构图在CPU上执行该处理函数。

您可以在CANN软件安装后文件存储路径下的“python/site-packages/dataflow/flow_func/flow_func.py”查看对应接口的实现。接口列表如下。

FlowMsg类

表1 FlowMsg类接口
接口名称	简介
FlowMsg构造函数	FlowMsg的构造函数。
get_msg_type	获取FlowMsg的消息类型。
get_tensor	获取FlowMsg中的tensor对象。
set_ret_code	设置FlowMsg消息中的错误码。
get_ret_code	获取输入FlowMsg消息中的错误码。
set_start_time	设置FlowMsg消息头中的开始时间戳。
get_start_time	获取FlowMsg消息中的开始时间戳。
set_end_time	设置FlowMsg消息头中的结束时间戳。
get_end_time	获取FlowMsg消息中的结束时间戳。
set_flow_flags	设置FlowMsg消息头中的flags。
get_flow_flags	获取FlowMsg消息头中的flags。
set_route_label	设置路由的标签。
get_transaction_id	获取FlowMsg消息中的事务ID，事务Id从1开始计数，每feed一批数据，事务Id会加一，可用于识别哪一批数据。

Tensor类

表2 Tensor类接口
接口名称	简介
Tensor构造函数和析构函数	Tensor构造函数和析构函数。
get_shape	获取Tensor的Shape。
get_data_type	获取Tensor中的数据类型。
get_data_size	获取Tensor中的数据大小。
get_element_cnt	获取Tensor中的元素的个数。
reshape	对tensor进行Reshape操作，不改变tensor的内容。

MetaParams类

表3 MetaParams类接口
接口名称	简介
PyMetaParams构造函数	PyMetaParams构造函数。
get_name	获取Flowfunc的实例名。
get_attr_int	获取指定名称的int类型属性值。
get_attr_bool_list	获取指定名称的bool数组类型属性值。
get_attr_int_list	获取指定名称的int数组类型属性值。
get_attr_int_list_list	获取指定名称的int二维数组类型属性值。
get_attr_bool	获取指定名称的bool类型属性值。
get_attr_float_list	获取指定名称的float数组类型属性值。
get_attr_tensor_dtype	获取指定名称的numpy dtype类型的属性值。
get_attr_tensor_dtype_list	获取指定名称的numpy dtype数组类型的属性值。
get_attr_str	获取指定名称的string类型的属性值。
get_attr_str_list	获取指定名称的string数组类型的属性值。
get_attr_float	获取指定名称的float类型属性值。
get_input_num	获取Flowfunc的输入个数。
get_output_num	获取Flowfunc的输出个数。
get_work_path	获取Flowfunc的工作路径。
get_running_device_id	获取正在运行的设备ID。

MetaRunContext类

表4 MetaRunContext类接口
接口名称	简介
MetaRunContext构造函数	MetaRunContext构造函数。
alloc_tensor_msg	根据shape、data type以及对齐大小申请tensor类型的FlowMsg。
set_output	设置指定index的output的tensor。
set_multi_outputs	批量设置指定index的output的tensor。
run_flow_model	同步执行指定的模型。
alloc_empty_data_msg	申请空数据的MsgType类型的message。
get_user_data	获取用户定义数据。
raise_exception	UDF主动上报异常。
get_exception	UDF获取其他UDF节点上报的异常。

AffinityPolicy类

表5 AffinityPolicy类接口
接口名称	简介
AffinityPolicy类	亲和策略枚举定义。

BalanceConfig类

表6 BalanceConfig类接口
接口名称	简介
BalanceConfig构造函数	BalanceConfig构造函数。
set_data_pos	设置输出数据对应权重矩阵中的位置。
get_inner_config	获取内部配置对象，被set_output或者set_multi_outputs调用。

UDF日志接口

表7 UDF日志接口
接口名称	简介
FlowFuncLogger构造函数	FlowFuncLogger构造函数。
get_log_header	获取日志扩展头信息。
is_log_enable	查询对应级别和类型的日志是否开启。
运行日志Error级别日志宏	运行日志Error级别日志宏。
运行日志Info级别日志宏	运行日志Info级别日志宏。
调试日志Error级别日志宏	调试日志Error级别日志宏。
调试日志Warn级别日志宏	调试日志Warn级别日志宏。
调试日志Info级别日志宏	调试日志Info级别日志宏。
调试日志Debug级别日志宏	调试日志Debug级别日志宏。

父主题： UDF API（Python）