简介
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]) |
父主题: Session