GetCapturedTensor
产品支持情况
产品
|
是否支持
|
|---|
Atlas A3 训练系列产品/Atlas A3 推理系列产品
|
√
|
Atlas A2 训练系列产品/Atlas A2 推理系列产品
|
√
|
头文件/库文件
- 头文件：#include <ge/fusion/match_result.h>
- 库文件：libge_compiler.so
功能说明
根据Pattern中捕获的索引，获取匹配中对应的Node output。
函数原型
Status GetCapturedTensor(size_t capture_idx, NodeIo &node_output) const
参数说明
参数名
|
输入/输出
|
说明
|
|---|
capture_idx
|
输入
|
捕获的索引ID。
|
node_output
|
输出
|
配到的Tensor。
|
返回值说明
参数名
|
类型
|
说明
|
|---|
-
|
Status
|
SUCCESS：成功。
FAILED：失败 。
|
约束说明
无