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 re模块介绍
Nov 30 Python
Python最火、R极具潜力 2017机器学习调查报告
Dec 11 Python
python: line=f.readlines()消除line中\n的方法
Mar 19 Python
Django model反向关联名称的方法
Dec 15 Python
Django ManyToManyField 跨越中间表查询的方法
Dec 18 Python
详解从Django Rest Framework响应中删除空字段
Jan 11 Python
Python实战之制作天气查询软件
May 14 Python
详解Python Opencv和PIL读取图像文件的差别
Dec 27 Python
PyCharm License Activation激活码失效问题的解决方法(图文详解)
Mar 12 Python
keras中的History对象用法
Jun 19 Python
如何基于Python实现word文档重新排版
Sep 29 Python
pytorch中的torch.nn.Conv2d()函数图文详解
Feb 28 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 咖啡文化
php 正则匹配函数体
2009/08/25 PHP
探讨Hessian在PHP中的使用分析
2013/06/13 PHP
PHP输入输出流学习笔记
2015/05/12 PHP
PHP实现简单用户登录界面
2019/10/23 PHP
AspNet中使用JQuery boxy插件的确认框
2015/05/20 Javascript
javascript从作用域链谈闭包
2020/07/29 Javascript
jQuery实现下拉加载功能实例代码
2016/04/01 Javascript
JavaScript中捕获与冒泡详解及实例
2017/02/03 Javascript
vue时间格式化实例代码
2017/06/13 Javascript
bootstrap table实现x-editable的行单元格编辑及解决数据Empty和支持多样式问题
2017/08/10 Javascript
关于meta viewport中target-densitydpi属性详解(推荐)
2017/08/18 Javascript
微信开发之企业付款到银行卡接口开发的示例代码
2018/09/18 Javascript
图片文字识别(OCR)插件Ocrad.js教程
2018/11/26 Javascript
npm 语义版本控制详解
2019/09/10 Javascript
小程序使用watch监听数据变化的方法详解
2019/09/20 Javascript
LayUI switch 开关监听 获取属性值、更改状态的方法
2019/09/21 Javascript
构建Vue大型应用的10个最佳实践(小结)
2019/11/07 Javascript
详解python单例模式与metaclass
2016/01/15 Python
Python with用法:自动关闭文件进程
2019/07/10 Python
python制作英语翻译小工具代码实例
2019/09/09 Python
Python使用Pandas读写Excel实例解析
2019/11/19 Python
python 实现表情识别
2020/11/21 Python
Python爬虫模拟登陆哔哩哔哩(bilibili)并突破点选验证码功能
2020/12/21 Python
Scrapy实现模拟登录的示例代码
2021/02/21 Python
Pytorch之扩充tensor的操作
2021/03/04 Python
css3闪亮进度条效果实现思路及代码
2013/04/17 HTML / CSS
html+css实现自定义图片上传按钮功能
2019/09/04 HTML / CSS
高中生家长会演讲稿
2014/01/14 职场文书
党员干部承诺书
2014/03/25 职场文书
中学生英语演讲稿
2014/04/26 职场文书
法学专业求职信
2014/07/15 职场文书
大三学年自我鉴定范文(3篇)
2014/09/28 职场文书
2016党校学习心得体会
2016/01/07 职场文书
ConstraintValidator类如何实现自定义注解校验前端传参
2021/06/18 Java/Android
MySQL事务操作的四大特性以及并发事务问题
2022/04/12 MySQL