CANN/ge GE Python Graph API

Graph

【免费下载链接】geGE（Graph Engine）是面向昇腾的图编译器和执行器，提供了计算图优化、多流并行、内存复用和模型下沉等技术手段，加速模型执行效率，减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力，并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge

产品支持情况

模块导入

from ge.graph import Graph

功能说明

Graph 类是 GE Python 接口的核心图操作类，用于管理计算图的构建、查询和修改。主要提供以下能力：

• 图生命周期管理：创建、销毁计算图，支持图的序列化与反序列化。
• 节点查询：获取图中所有节点或直接节点，按名称查找节点。
• 属性管理：获取和设置图级别的自定义属性。
• 边操作：添加/删除数据边和控制边，构建节点间的数据依赖和控制依赖关系。
• 子图管理：添加、查询、删除子图，支持层级化的图结构。
• 序列化与持久化：将图导出为文件（dump_to_file）或字符串（dump_to_stream），支持保存和加载 AIR 格式模型。

函数原型

__init__

def __init__(self, name: Optional[str] = "graph") -> None

创建一个 Graph 对象。

name（属性）

@property def name(self) -> str

获取图名称。

get_all_nodes

def get_all_nodes(self) -> List[Node]

获取图中所有节点，包括子图中的节点。

get_direct_nodes

def get_direct_nodes(self) -> List[Node]

获取当前图的直接节点，不包含子图中的节点。

get_attr

def get_attr(self, key: str) -> Any

获取图的指定属性值。

set_attr

def set_attr(self, key: str, value: Any) -> None

设置图的指定属性。

dump_to_file

def dump_to_file(self, format: DumpFormat = DumpFormat.kReadable, suffix: str = "") -> None

将图导出到文件。

dump_to_stream

def dump_to_stream(self, format: DumpFormat = DumpFormat.kReadable) -> str

将图导出为字符串。

save_to_air

def save_to_air(self, file_path: str) -> None

将图保存为 AIR 格式文件。

load_from_air

def load_from_air(self, file_path: str) -> None

从 AIR 格式文件加载图。

remove_node

def remove_node(self, node: Node) -> None

从图中移除指定节点。

remove_edge

def remove_edge(self, src_node: Node, src_port_index: int, dst_node: Node, dst_port_index: int) -> None

移除指定的边。

add_data_edge

def add_data_edge(self, src_node: Node, src_port_index: int, dst_node: Node, dst_port_index: int) -> None

添加数据边。

add_control_edge

def add_control_edge(self, src_node: Node, dst_node: Node) -> None

添加控制边。

find_node_by_name

def find_node_by_name(self, name: str) -> Node

根据节点名称查找节点。

get_all_subgraphs

def get_all_subgraphs(self) -> List[Graph]

获取图中所有子图。

get_subgraph

def get_subgraph(self, name: str) -> Optional[Graph]

根据名称获取指定子图。

add_subgraph

def add_subgraph(self, subgraph: Graph) -> None

向图中添加子图。

remove_subgraph

def remove_subgraph(self, name: str) -> None

根据名称移除子图。

参数说明

__init__

name

无参数（只读属性）。

get_all_nodes

无参数。

get_direct_nodes

无参数。

get_attr

set_attr

dump_to_file

DumpFormat 枚举值说明：

dump_to_stream

save_to_air

load_from_air

remove_node

remove_edge

add_data_edge

add_control_edge

find_node_by_name

get_all_subgraphs

无参数。

get_subgraph

add_subgraph

remove_subgraph

返回值说明

约束说明

• 所有权模型：Graph 对象存在两种所有权状态。默认情况下由 Python 侧管理 C++ 资源的生命周期。当 Graph 作为子图参数传递给算子（如 If、While、Case）时，所有权会自动转移至 C++ 侧，以避免双重释放问题。
• 禁止拷贝：Graph 类不支持拷贝操作（浅拷贝和深拷贝均不支持），调用copy或deepcopy将抛出 RuntimeError。
• 子图名称唯一性：调用add_subgraph添加子图时，子图名称在父图中必须唯一。若名称已存在，操作将失败并抛出异常。
• 边的端口索引：调用remove_edge移除控制边时，src_port_index和dst_port_index均应设置为 -1。移除数据边时，端口索引必须与实际连接的端口一致。
• dump_to_file 输出限制：使用 DumpFormat.kOnnx 格式导出时，pbtxt 文件仅包含图结构信息，不包含权重数据或其他属性。
• 类型校验：所有方法的参数均进行类型校验，类型不匹配时将抛出 TypeError；操作失败时将抛出 RuntimeError。
• 节点查找：find_node_by_name在未找到指定名称的节点时会抛出 RuntimeError，而非返回 None。使用前需确认节点确实存在。

【免费下载链接】geGE（Graph Engine）是面向昇腾的图编译器和执行器，提供了计算图优化、多流并行、内存复用和模型下沉等技术手段，加速模型执行效率，减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力，并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge