未能应用委托：TfLiteGpu委托初始化：MUL：期望形状为HxWxC的3D张量或形状为1xHxWxC的4D张量，但得到98x8（Android）

Question

我从2.3.0升级到2.4.0，并在初始化过程中得到这个错误

java.lang.IllegalArgumentException：内部错误：未能应用委托：TfLiteGpu委托初始化：MUL：期望形状为HxWxC的3D张量或形状为1xHxWxC的4D张量，但得到98x8 TfLiteGpu委托准备：委托未初始化节点编号329（TfLiteGpu委托V2）未能准备。

val tfliteOptions = Interpreter
.Options()
.setNumThreads(THREADS_COUNT)
.addDelegate(GpuDelegate())

Interpreter(loadModelFile(context), tfliteOptions)

问题出在tenorflow lite-gpu 2.4.0中的某个地方

如果我使用这样的配置与旧版本都工作得很好

implementation("org.tensorflow:tensorflow-lite:2.4.0")
implementation("org.tensorflow:tensorflow-lite-gpu:2.3.0")

2.4.0、2.5.0、2.6.0、2.7.0、2.8.0 Gpu代理不工作

GitHub上的问题https://github.com/tensorflow/tensorflow/issues/45845

示例android项目如何复制https://github.com/OleksandrGrument/GpuDelegateBugReproduce

Tflite模型https://github.com/OleksandrGrument/GpuDelegateBugReproduce/blob/master/app/src/main/assets/model.tflite

我在github上打开了一个线程，但两年来没有结果，也许有人知道如何通过更改tflite文件来绕过这个错误

Answer 1

在转换为我更改的tflite之前，我找到了解决此问题的方法

tf.math.multiply(weights, var)

到

tf.linalg.matmul(weights, var, transpose_a=True)

看起来，从2.4.0版开始，Gpu委托无法继续此操作以将2d数组与tf.math.multiply相乘。