am15h / tflite_flutter_plugin

TensorFlow Lite Flutter Plugin
https://pub.dev/packages/tflite_flutter
Apache License 2.0
508 stars 353 forks source link

Output error #98

Closed Srikarthikeyan4006 closed 3 years ago

Srikarthikeyan4006 commented 3 years ago

Total elements mismatch expected: 1 elements for shape: [1, 1] but found 2

output decleration var output = List(2).reshape([1,1]);

am15h commented 3 years ago

@Srikarthikeyan4006 What is the target shape that you are trying to achieve? If your target shape is [a, b, c] then you should be doing. var output = List(a*b*c).reshape([a, b, c]);