为 runForMultipleInputsOutputs 分配内存时发生

Question

我在为 Android Java 的 TFLITE 解释器准备输出时遇到错误。该模型有 1 个输入和 4 个输出。interpreter.runForMultipleInputsOutputs(input, map_of_indices_to_outputs);E/Run multiple: Internal error: Unexpected failure when preparing tensor allocations: tensorflow/lite/kernels/tile.cc:53 num_dimensions != num_multipliers (1 != 2)Node number 4 (TILE) failed to prepare.输出要求是 4 个浮点数组的列表：[ [1x1],[1x2], [1x2], [1x2] ]python 中预测的输出是：In [56] output = model.predict(new_observation_scaled)Out[56]: [array([[0.]], dtype=float32), array([[137.66626, 335.7148 ]], dtype=float32), array([[0.16666616, 0.40643442]], dtype=float32), array([[9.9915421e-01, 8.4577635e-04]], dtype=float32)]所以我用JAVA准备了一个对象列表：float [][] output0 = new float [1][1];float [][] output1 = new float [1][2];float [][] output2 = new float [1][2];float [][] output3 = new float [1][2];Object[] outputs = {output0,output1,output2,output3};Map<Integer, Object> map_of_indices_to_outputs = new HashMap<>();map_of_indices_to_outputs.put(0, output0);map_of_indices_to_outputs.put(1, output1);map_of_indices_to_outputs.put(2, output2);map_of_indices_to_outputs.put(3, output3);你能帮我找出错误吗？

Answer 1

我得到了问题的解决方案：

以同样的方式，输出需要一个列表，输入也需要一个列表：

Object[] outputs = {output0,output1,output2,output3};

Object[] inputs = {input};

interpreter.runForMultipleInputsOutputs(inputs, map_of_indices_to_outputs);