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使用在线API查询IP对应的地理位置信息实例
Jun 01 Python
Python实现远程调用MetaSploit的方法
Aug 22 Python
python基础知识小结之集合
Nov 25 Python
再谈Python中的字符串与字符编码(推荐)
Dec 14 Python
Python、PyCharm安装及使用方法(Mac版)详解
Apr 28 Python
Python多线程爬虫实战_爬取糗事百科段子的实例
Dec 15 Python
Python反射用法实例简析
Dec 22 Python
树莓派用python中的OpenCV输出USB摄像头画面
Jun 22 Python
Python如何避免文件同名产生覆盖
Jun 09 Python
Python下划线5种含义代码实例解析
Jul 10 Python
Python中logger日志模块详解
Aug 04 Python
Python图片验证码降噪和8邻域降噪
Aug 30 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
咖啡语言
2021/03/03 咖啡文化
Ajax PHP 边学边练 之三 数据库
2009/11/26 PHP
PHP运行时强制显示出错信息的代码
2011/04/20 PHP
PHP使用反射机制实现查找类和方法的所在位置
2016/04/22 PHP
WordPress中的shortcode短代码功能使用详解
2016/05/17 PHP
在Laravel5.6中使用Swoole的协程数据库查询
2018/06/15 PHP
在Laravel的Model层做数据缓存的实现
2019/09/26 PHP
jQuery的DOM操作之删除节点示例
2014/01/03 Javascript
Javascript模拟加速运动与减速运动代码分享
2014/12/11 Javascript
js中取得变量绝对值的方法
2015/01/03 Javascript
javascript实现表单提交后,提交按钮不可用的方法
2015/04/18 Javascript
DEDECMS如何为文章添加HOT NEW标志图片
2015/08/14 Javascript
JavaScript中的this使用详解
2016/07/27 Javascript
layui获取多选框中的值方法
2018/08/15 Javascript
vue axios数据请求get、post方法及实例详解
2018/09/11 Javascript
在Vue-cli里应用Vuex的state和mutations方法
2018/09/16 Javascript
Node.js的进程管理的深入理解
2019/01/09 Javascript
javscript 数组扁平化的实现
2020/02/03 Javascript
JS获取当前时间戳方法解析
2020/08/29 Javascript
Django发送邮件和itsdangerous模块的配合使用解析
2019/08/10 Python
使用python绘制温度变化雷达图
2019/10/18 Python
python和c语言哪个更适合初学者
2020/06/22 Python
Python如何执行精确的浮点数运算
2020/07/31 Python
python如何实现图片压缩
2020/09/11 Python
如何利用Python给自己的头像加一个小国旗(小月饼)
2020/10/02 Python
CSS3的文字阴影—text-shadow的使用方法
2012/12/25 HTML / CSS
国际知名设计师时装商店:Coggles
2016/09/05 全球购物
计算机应用专业应届毕业生中文求职信范文
2013/11/29 职场文书
《赠汪伦》教学反思
2014/04/12 职场文书
三严三实对照检查材料
2014/09/22 职场文书
关于运动会广播稿300字
2014/10/05 职场文书
支教个人总结
2015/03/04 职场文书
社区义诊通知
2015/04/24 职场文书
捐款通知怎么写
2015/04/24 职场文书
详解Java实践之适配器模式
2021/06/18 Java/Android
了解Kubernetes中的Service和Endpoint
2022/04/01 Servers