get_input_tensors - 3.0 简体中文

Vitis AI 用户指南 (UG1414)

Document ID
UG1414
Release Date
2023-02-24
Version
3.0 简体中文

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

原型设计

List[xir.Tensor] get_input_tensors()

参数

返回

DPU 运行器输入列表,每个输入类型均为 xir.Tensor。

用法

get_input_tensors() 返回的列表的每个元素都与一个 DPU 运行器输入相对应。每个列表元素都具有若干个类属性,用法如下所示:
inputTensors = dpu_runner.get_input_tensors()
print(dir(inputTensors[0])
这些属性中最实用的有 name、dims 和 dtype:
for inputTensor in inputTensors:
print(inputTensor.name)
print(inputTensor.dims)
print(inputTensor.dtype)
请注意,输入张量的维度 (.dim) 采用 NHWC 格式(批次大小、高度、宽度、通道数)。