昇腾社区首页
中文
注册

FlowNode

产品支持情况

产品

是否支持

Atlas A3 训练系列产品/Atlas A3 推理系列产品

x

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

Atlas 200I/500 A2 推理产品

x

Atlas 推理系列产品

x

Atlas 训练系列产品

x

函数功能

DataFlow Graph中的计算节点。

函数原型

1
FlowNode(input_num, output_num, name=None)

参数说明

参数名称

数据类型

取值说明

input_num

int

节点的输入个数。

output_num

int

节点的输出个数。

name

str

节点名称,框架会自动保证名称唯一,不设置时会自动生成FlowNode, FlowNode_1, FlowNode_2,...的名称。

返回值

正常场景下返回None。

返回“TypeError”表示参数类型不正确。

调用示例

1
2
import dataflow as df
flow_node = df.FlowNode(input_num=2, output_num=1)

约束说明