get_inputs - 3.0 简体中文

获取 RunnerExt 的所有输入 TensorBuffer。

原型设计

List[vart.TensorBuffer] get_inputs()

参数

无。

返回

所有输入 TensorBuffer。指向输入 TensorBuffer 的原始指针的矢量。

用法

input_tensor_buffers = runner.get_inputs()
input_dim = tuple(input_tensor_buffers[0].get_tensor().dims)