昇腾社区首页
中文
注册
开发者
下载

GetCapturedTensor

产品支持情况

产品

是否支持

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

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

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

头文件/库文件

  • 头文件:#include <ge/fusion/match_result.h>
  • 库文件:libge_compiler.so

功能说明

根据Pattern中捕获的索引,获取匹配中对应的Node output。

函数原型

1
Status GetCapturedTensor(size_t capture_idx, NodeIo &node_output) const

参数说明

参数名

输入/输出

说明

capture_idx

输入

捕获的索引ID。

node_output

输出

配到的Tensor。

返回值说明

参数名

类型

说明

-

Status

SUCCESS:成功。

FAILED:失败 。

约束说明