get_output_tensors - 3.0 简体中文

获取运行器的所有输出张量。

原型设计

List[xir.Tensor] get_output_tensors()

参数

无

返回

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

用法

outputTensors = runner.get_output_tensors()
shapeOut = tuple(outputTensors[0].dims)