简介

from ge.session import Session
from ge.graph import Tensor
from ge.ge_global import GeApi

# 初始化 GE
config = {
    "ge.execDeviceId": "0",
    "ge.graphRunMode": "0"
}
GeApi.ge_initialize(config)


# 创建会话
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])

# 终结 GE
GeApi.ge_finalize()