pytorch模型预测结果与ndarray互转方式


Posted in Python onJanuary 15, 2020

预测结果转为numpy:

logits=model(feature)
 
#如果模型是跑在GPU上
result=logits.data.cpu().numpy()  /  logits.cpu().numpy()
 
#如果模型跑在cpu上
result=logits.data.numpy()  /  logits.numpy()

将矩阵转为tensor:

np_arr = np.array([1,2,3,4])
tensor=torch.from_numpy(np_arr)

以上这篇pytorch模型预测结果与ndarray互转方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
利用Python的Django框架生成PDF文件的教程
Jul 22 Python
不要用强制方法杀掉python线程
Feb 26 Python
详解Django中间件执行顺序
Jul 16 Python
Django给admin添加Action的步骤详解
May 01 Python
PyCharm-错误-找不到指定文件python.exe的解决方法
Jul 01 Python
详解python实现交叉验证法与留出法
Jul 11 Python
Numpy数组array和矩阵matrix转换方法
Aug 05 Python
Django使用uwsgi部署时的配置以及django日志文件的处理方法
Aug 30 Python
opencv3/python 鼠标响应操作详解
Dec 11 Python
vue学习笔记之动态组件和v-once指令简单示例
Feb 29 Python
使用python处理题库表格并转化为word形式的实现
Apr 14 Python
Python数据分析之pandas函数详解
Apr 21 Python
pytorch实现对输入超过三通道的数据进行训练
Jan 15 #Python
Pytorch 定义MyDatasets实现多通道分别输入不同数据方式
Jan 15 #Python
pytorch构建多模型实例
Jan 15 #Python
利用Pytorch实现简单的线性回归算法
Jan 15 #Python
pytorch实现线性拟合方式
Jan 15 #Python
Python 支持向量机分类器的实现
Jan 15 #Python
pytorch-神经网络拟合曲线实例
Jan 15 #Python
You might like
PHP的宝库目录--PEAR
2006/10/09 PHP
php生成随机密码自定义函数代码(简单快速)
2014/05/10 PHP
php中使用GD库做验证码
2016/03/31 PHP
超强多功能php绿色集成环境详解
2017/01/25 PHP
php中get_magic_quotes_gpc()函数说明
2017/02/06 PHP
Jquery的hide及toggle方法让超链接慢慢消失
2013/09/06 Javascript
表单元素与非表单元素刷新区别详细解析
2013/11/06 Javascript
原生JavaScript实现瀑布流布局
2020/06/28 Javascript
有关JavaScript中call()和apply() 的一些理解
2016/05/20 Javascript
基于jQuery实现页面搜索功能
2020/03/26 Javascript
AngularJs定制样式插入到ueditor中的问题小结
2016/08/01 Javascript
Jquery根据浏览器窗口改变调整大小的方法
2017/02/07 Javascript
javascript实现延时显示提示框效果
2017/06/01 Javascript
Vue实现搜索 和新闻列表功能简单范例
2018/03/16 Javascript
微信小程序适配iphoneX的实现方法
2018/09/18 Javascript
微信小程序自定义弹窗wcPop插件
2018/11/19 Javascript
vue子路由跳转实现tab选项卡
2019/07/24 Javascript
vue实现前端列表多条件筛选
2020/10/26 Javascript
[03:17]DOTA2英雄基础教程 剧毒术士
2013/12/12 DOTA
python Web开发你要理解的WSGI & uwsgi详解
2018/08/01 Python
pandas的排序和排名的具体使用
2019/07/31 Python
python创建学生管理系统
2019/11/22 Python
Python unittest框架操作实例解析
2020/04/13 Python
Python列表去重复项的N种方法(实例代码)
2020/05/12 Python
python产生模拟数据faker库的使用详解
2020/11/04 Python
python实现图像随机裁剪的示例代码
2020/12/10 Python
博朗(Braun)俄罗斯官方商店:德国小家电品牌
2019/09/24 全球购物
双立人加拿大官网:Zwilling加拿大
2020/08/10 全球购物
简述安装Slackware Linux系统的过程
2012/01/12 面试题
排序都有哪几种方法?请列举。用JAVA实现一个快速排序
2014/02/16 面试题
写求职信有什么意义
2014/02/17 职场文书
中国入世承诺
2014/04/01 职场文书
竞聘演讲稿
2014/04/24 职场文书
2015元旦晚会主持人开场白+结束语
2014/12/14 职场文书
主持人开幕词
2015/01/29 职场文书
绍兴鲁迅故居导游词
2015/02/09 职场文书