wc591723150
鬼才教头泰伦卢
2020-07-15 21:51

pytorch移植到安卓,模型有多个输出怎么办

5
  • android-studio
  • 深度学习
  • android
  • 人工智能

使用官方的方法将模型保存为.pt文件,然后在安卓编程时使用Tensor outputTensor = mModule.forward(IValue.from(inputTensor)).toTensor()得到模型的输出,但是官方演示的model输出应该就是一个tensor,可是我的model输出是3个tensor,应该怎么得到这三个tensor呢?

  • 点赞
  • 回答
  • 收藏
  • 复制链接分享

6条回答

为你推荐

换一换