Android NN API CpuExecutor: неверный тип или размеры для вывода модели

Question

Я пытаюсь реализовать простую двухслойную сеть в Android NN API, но во время компиляции выдается ошибка:

E / CpuExecutor: Неверный тип или размеры для вывода модели Ошибка SOFTMAX.

E / CpuExecutor: Неверный тип или размеры для вывода модели Ошибка SOFTMAX.

может кто знает с чем связана эта ошибка? P / S / Если у кого есть рабочий пример реализации в NN API в репо? был бы очень признателен

Answer 1

Проблема решена, размер выходного тензора в предыдущем слое не совпадает. Если у кого-нибудь есть пример встроенной модели для классификации изображения в NNAPI, я буду благодарен.