CaptureTensor
产品支持情况
产品
|
是否支持
|
|---|
Atlas A3 训练系列产品/Atlas A3 推理系列产品
|
√
|
Atlas A2 训练系列产品/Atlas A2 推理系列产品
|
√
|
头文件/库文件
- 头文件：#include <ge/fusion/pattern.h>
- 库文件：libge_compiler.so
功能说明
捕获Pattern中一个Tensor，从而在Match Result中可以按序获取。
捕获Node的output，用于后续在Match Result中通过捕获时的索引，快速获取到Node output，用户可以从Match Result中拿到Tensor描述做校验。
函数原型
Pattern &CaptureTensor(const NodeIo &node_output)
参数说明
参数名
|
输入/输出
|
说明
|
|---|
node_output
|
输入
|
Tensor的来源，即来自某个Node的某个输出。
因node_output可唯一标定一个图中Tensor，因此常用node_output指代Tensor。
|
返回值说明
参数名
|
类型
|
说明
|
|---|
-
|
Pattern &
|
Pattern的引用，便于级联调用。
|
约束说明
无