之后图形和库参数的差异继电器.u模块.build()

2024-09-28 17:25:10 发布

您现在位置:Python中文网/ 问答频道 /正文

在TVM库中,当使用_relay.build_module.build()_时,relay.buildtvm.build之后的输出结果是什么

作为输出的两个参数(graph和library)的区别和需要是什么?这两个参数后来用于运行图形或创建图形运行时?你知道吗

例如,在本代码中:

opt_level = 3
target = tvm.target.cuda()
with relay.build_config(opt_level=opt_level):
    graph, lib, params = relay.build_module.build(
        net, target, params=params)

什么是输出图形? 我们可以只使用其中的一个来生成图形运行时,还是必须像下面这样同时使用这两个:

# create random input
ctx = tvm.gpu()
data = np.random.uniform(-1, 1, size=data_shape).astype("float32")
# create module
module = graph_runtime.create(graph, lib, ctx)
# set input and parameters
module.set_input("data", data)
module.set_input(**params)
# run
module.run()
# get output
out = module.get_output(0, tvm.nd.empty(out_shape)).asnumpy()

Tags: build图形targetinputdata参数createparams