TFLite interpreter.get_input_details中引用的“索引”是什么?

时间:2019-08-28 07:39:53

标签: tensorflow keras tensorflow-lite

我正在对Keras / TensorFlow模型进行一些推断,但是文档显示似乎有点稀疏,所以我正在尝试尽可能多地学习和记录文档,而不仅仅是依赖复制的代码示例。示例包括以下行:

interpreter.get_input_details()

哪个返回包含单个字典的列表,其中包含以下信息:

[{'name': 'conv2d_input', 'index': 8, 'shape': array([ 1, 28, 28, 1]), 'dtype': <class 'numpy.float32'>, 'quantization': (0.0, 0)}]

get_output_details返回的内容非常相似。其中大多数是不言自明的,但我不清楚该索引指的是什么。我发现的示例表明set_tensor和get_tensor方法需要该数字。到目前为止,我仅在某处发表评论,说这是张量索引,对我而言似乎是重言式!

谢谢。

1 个答案:

答案 0 :(得分:0)

在TFLite解释器中,所有张量都放入张量列表中(请参见TfLiteContext中的TfLiteTensor* tensors;),索引是张量列表中张量的索引。