CaptureTensor

Applicability

Product	Supported or Not
Atlas A3 training products/Atlas A3 inference products	√
Atlas A2 training products/Atlas A2 inference products	√
Atlas 200I/500 A2 inference products	√
Atlas inference products	√
Atlas training products	√

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 match results.

It captures the output of a node, enabling quick access to the node's output by index in the matching later. You can also retrieve the tensor description from the matching 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 a Pattern object, used to support cascading method calls.

Constraints

None

Parent topic: Pattern