get_input_tensors - 2.5 日本語

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 (バッチ サイズ、高さ、幅、チャネル数) 形式です。