昇腾社区首页
中文
注册
开发者
下载

简介

Session类用于管理图的执行会话。

会话执行调用示例:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
from ge.session import Session
from ge.graph import Tensor

# 创建会话
session = Session()

# 添加图
session.add_graph(0, graph)

# 准备输入
input_tensor = Tensor(data=[1.0, 2.0, 3.0, 4.0], data_type=DataType.DT_FLOAT, format=Format.FORMAT_ND, shape=[2, 2])

# 运行图
outputs = session.run_graph(0, [input_tensor])