Android NN API CpuExecutor:模型输出的类型或尺寸无效

时间:2018-11-09 15:17:06

标签: android neural-network java-native-interface nnapi

我正在尝试在Android NN API中实现一个简单的两层网络,但是在编译过程中出现错误:

  

E / CpuExecutor:模型输出的类型或尺寸无效       SOFTMAX失败。

     

E / CpuExecutor:模型输出的类型或尺寸无效       SOFTMAX失败。

也许谁知道此错误与什么有关? P / S /是否有人在回购中提供了在NN API中实现的可行示例?会非常感激的

1 个答案:

答案 0 :(得分:0)

问题已解决,上一层中输出张量的大小不匹配。如果有人提供了用于在NNAPI中对图像进行分类的构建模型的示例,我将不胜感激。