pytorch
关注
分享
扫一扫
西伯利亚的蓝眼睛
这个作者很懒，什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
如何在c++中使用pytorch模型
同步知乎页：https://zhuanlan.zhihu.com/p/66707105该过程有两步：第一：在python中导出pytorch模型第二：在c++中加载并执行第一步：convert Pytorch func/model to Torch Script有2中方法可以完成该任务。一、 torch.jit.trace function def foo...原创 2019-05-23 10:33:31 · 1679 阅读 · 3 评论 -
c++ load pytorch 的数据转换
tensor、numpy、vector转换python中：**numpy -> tensor**: `torch.from_numpy(ndarray)` **tensor -> numpy**: `tensor.numpy()`c++中：**array -> tensor**: `torch::tensor(at::ArrayRef<float&...原创 2019-05-24 10:42:13 · 9738 阅读 · 16 评论