gpt4 book ai didi

tensorflow - TFLite interpreter.get_input_details 中的 'index' 指的是什么?

转载 作者:行者123 更新时间:2023-12-04 13:00:47 25 4
gpt4 key购买 nike

我正在对 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 方法需要该数字。到目前为止我发现的只是某处的评论说这是指张量指数,对我来说似乎是同义反复!

提前致谢。

最佳答案

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

关于tensorflow - TFLite interpreter.get_input_details 中的 'index' 指的是什么?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/57687251/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com