CaptureTensor

Applicability

Product	Supported
Atlas 350 Accelerator Card	√
Atlas A3 training product/Atlas A3 inference product	√
Atlas A2 training product/Atlas A2 inference product	√
Atlas 200I/500 A2 inference product	√
Atlas inference product	√
Atlas training product	√

Header File/Library File

Header file: #include <ge/fusion/pattern.h>
Library file: libge_compiler.so

Function Usage

Captures a tensor within a pattern, allowing it to be retrieved in order from the MatchResult.

It captures the output of a node, enabling fast retrieval of the node's output via the index assigned during capture in subsequent matching. You can also retrieve the tensor description from the match results for verification.

Prototype

Pattern &CaptureTensor(const NodeIo &node_output)

Parameters

Parameter	Input/Output	Description
node_output	Input	Tensor source, that is, an output of a node. As node_output can uniquely identify a tensor in a graph, it is often used to represent a tensor.

Parameter

Input/Output

Description

node_output

Input

Tensor source, that is, an output of a node.

As node_output can uniquely identify a tensor in a graph, it is often used to represent a tensor.

Returns

Parameter	Type	Description
-	Pattern &	Reference to the pattern for cascading calls.

Restrictions

None

Parent topic: Pattern