TensorflowLite java.lang.IllegalArgumentException: Cannot copy between a TensorFlowLite tensor with shape [0, 1] and a Java object with shape [1, 1]

Использую tensorflow lite для конвертирования модели бинарной классификации. Выходной тензор имеет shape = [1,1]. Но после того как я передаю свой outputs, возникает ошибка. В отладке я увидел, что правильная форма меняется в методе this.outputTensors[i].refreshShape(); Он обновляет правильный выход [1,1] на [0,1]

java.lang.IllegalArgumentException: Cannot copy between a TensorFlowLite tensor with shape [0, 1] and a Java object with shape [1, 1]. at org.tensorflow.lite.Tensor.throwIfShapeIsIncompatible(Tensor.java:282) at org.tensorflow.lite.Tensor.throwIfDataIsIncompatible(Tensor.java:249) at org.tensorflow.lite.Tensor.copyTo(Tensor.java:141) at org.tensorflow.lite.NativeInterpreterWrapper.run(NativeInterpreterWrapper.java:161) at org.tensorflow.lite.Interpreter.runForMultipleInputsOutputs(Interpreter.java:275) at org.tensorflow.lite.Interpreter.run(Interpreter.java:249) at com.ml.quaterion.spamo.MainActivity.classifySequence(MainActivity.kt:72) at com.ml.quaterion.spamo.MainActivity$onCreate$1$1.onDataProcessed(MainActivity.kt:36) at com.ml.quaterion.spamo.Classifier$LoadVocabularyTask.onPostExecute(Classifier.kt:118) at com.ml.quaterion.spamo.Classifier$LoadVocabularyTask.onPostExecute(Classifier.kt:101) at android.os.AsyncTask.finish(AsyncTask.java:696) at android.os.AsyncTask.access$600(AsyncTask.java:180) at android.os.AsyncTask$InternalHandler.handleMessage(AsyncTask.java:713) at android.os.Handler.dispatchMessage(Handler.java:112) at android.os.Looper.loop(Looper.java:216) at android.app.ActivityThread.main(ActivityThread.java:7625) at java.lang.reflect.Method.invoke(Native Method) at com.android.internal.os.RuntimeInit$MethodAndArgsCaller.run(RuntimeInit.java:524) at com.android.internal.os.ZygoteInit.main(ZygoteInit.java:987)