使用官方的方法将模型保存为.pt文件,然后在安卓编程时使用Tensor outputTensor = mModule.forward(IValue.from(inputTensor)).toTensor()得到模型的输出,但是官方演示的model输出应该就是一个tensor,可是我的model输出是3个tensor,应该怎么得到这三个tensor呢?
打一下log 看看 这三个tensor的输出时间 是一起的还是 有时间间隔的
最简单的是分为3个模型,跑3次正向传播算法,就可以得到3个tensor了。
输出有三个tensor就三个tensor呀,你输出用三个变量去接收不就得了,要不然你就在model的最后一层,将三个tensor concat到一起,然后推断的时候再自己分割不就得了
楼主请问您的这个问题解决了么,我也遇到同样问题,希望能得到点拨,谢谢
我q1014657423
.toTuple()