鬼才教头泰伦卢
2020-07-15 21:51pytorch移植到安卓，模型有多个输出怎么办
5使用官方的方法将模型保存为.pt文件，然后在安卓编程时使用Tensor outputTensor = mModule.forward(IValue.from(inputTensor)).toTensor()得到模型的输出，但是官方演示的model输出应该就是一个tensor，可是我的model输出是3个tensor，应该怎么得到这三个tensor呢？
- 点赞
- 回答
- 收藏
- 复制链接分享
6条回答
为你推荐
- pytorch point源代码出现KeyError: Caught KeyError in DataLoader worker process 0.如何解决？
- 深度学习
- 机器学习
- 人工智能
- 1个回答
- 关于 pytorch中Tensor数据类型的使用问题
- 深度学习
- 人工智能
- 神经网络
- python
- 机器学习
- 1个回答
- pytorch cuda版运行出错 invalid start byte
- python
- 神经网络
- 2个回答
- 求助：CUDA的RuntimeError：cuda runtime error (30)
- python
- 神经网络
- 6个回答
- pytorch加载model发现key的值有差异，能不能修改
- 深度学习
- 神经网络
- python
- 机器学习
- 1个回答
换一换