DataType error: cannot resolve DataType of [[[D

问题描述：
在安卓里调用tflite模型时，运行到tfLite.run(input, output);这句就会报错。提示如题所示错误。

原因分析：
在安卓调用tflite模型时，模型输入与模型输出都要与在python里训练时的输入输出相对应。我的模型在python里的输入形式是[1, 32, 256, 1]，类型是float32，则在JAVA里也要是这种类型的ByteBuffer。
输出是[32,1,5116]，类型是float32，则在java里接收模型输出的变量也应该是float类型的，可为float output[][][] = new float[32][1][5116];写成double类型的就会报如题的错误，数组中后面两维的数字[1][5116]都要写全。