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