获取运行器的所有输入张量。
原型设计
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 格式(批次大小、高度、宽度、通道数)。