get_outputs - 3.0 简体中文

获取 RunnerExt 的所有输出 TensorBuffer。

原型设计

List[vart.TensorBuffer] get_outputs()

参数

无。

返回

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

用法

output_tensor_buffers = runner.get_outputs()
output_element_num = tuple(output_tensor_buffers[0].get_tensor().get_element_num())