SetOutput

产品支持情况

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

头文件/库文件

头文件：#include <ge/es_graph_builder.h>
库文件：libeager_style_graph_builder_base.so、libeager_style_graph_builder_base_static.a

功能说明

设置图的输出节点。

函数原型

static int64_t SetOutput(const EsTensorHolder &tensor, int64_t index);

参数说明

参数名	输入/输出	说明
tensor	输入	输出张量持有者。
index	输入	输出索引，从0开始计数。

返回值说明

参数名	类型	说明
-	int64_t	返回操作结果，成功为0，失败返回非0值。

约束说明

无

调用示例

// 1. 创建图构建器（EsGraphBuilder）
EsGraphBuilder builder("graph_name");
// 2. 添加 2 个输入节点
EsTensorHolder [data0, data1] = builder.CreateInputs<2>();
// 3. 添加中间节点，C++中，加减乘除等常用运算符被重载，可以直接使用
EsTensorHolder add = data0 + data1;
// 4. 设置图输出，add为第0个输出tensor
builder.SetOutput(add, 0);

父主题： EsGraphBuilder