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 相关文章推荐
以Flask为例讲解Python的框架的使用方法
Apr 29 Python
python解析xml文件实例分析
May 27 Python
Python多进程并发(multiprocessing)用法实例详解
Jun 02 Python
Python 逐行分割大txt文件的方法
Oct 10 Python
K-近邻算法的python实现代码分享
Dec 09 Python
Python读写zip压缩文件的方法
Aug 29 Python
在PyCharm中三步完成PyPy解释器的配置的方法
Oct 29 Python
Pycharm中Python环境配置常见问题解析
Jan 16 Python
jupyter notebook 参数传递给shell命令行实例
Apr 10 Python
scrapy爬虫:scrapy.FormRequest中formdata参数详解
Apr 30 Python
Python包资源下载路径报404解决方案
Nov 05 Python
Python之京东商品秒杀的实现示例
Jan 06 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中转义mysql语句的实现代码
2011/06/24 PHP
php文件夹与文件目录操作函数介绍
2013/09/09 PHP
PHP嵌套输出缓冲代码实例
2015/05/12 PHP
Laravel框架路由和控制器的绑定操作方法
2018/06/12 PHP
JS实现的一个简单的Autocomplete自动完成例子
2014/04/16 Javascript
JavaScript中原型和原型链详解
2015/02/11 Javascript
javascript实现百度地图鼠标滑动事件显示、隐藏
2015/04/02 Javascript
wangEditor编辑器失去焦点后仍然可以在原位置插入图片分析
2015/05/06 Javascript
jQuery满意度星级评价插件特效代码分享
2015/08/19 Javascript
基于JavaScript实现动态添加删除表格的行
2016/02/01 Javascript
微信开发 js实现tabs选项卡效果
2016/10/28 Javascript
laydate.js日期时间选择插件
2017/01/04 Javascript
js学习总结之dom2级事件基础知识详解
2017/07/27 Javascript
js变量值传到php过程详解 将php解析成数据
2019/06/26 Javascript
Vue-Cli项目优化操作的实现
2019/10/27 Javascript
vue中使用GraphQL的实例代码
2019/11/04 Javascript
加速vue组件渲染之性能优化
2020/04/09 Javascript
在Vue中使用CSS3实现内容无缝滚动的示例代码
2020/11/27 Vue.js
python实现DES加密解密方法实例详解
2015/06/30 Python
Python解析json文件相关知识学习
2016/03/01 Python
在CMD命令行中运行python脚本的方法
2018/05/12 Python
python判断数字是否是超级素数幂
2018/09/27 Python
新年快乐! python实现绚烂的烟花绽放效果
2019/01/30 Python
python3 pillow模块实现简单验证码
2019/10/31 Python
HTML5 canvas基本绘图之绘制矩形
2016/06/27 HTML / CSS
英国床和浴室商场:Bed & Bath Emporium
2018/05/20 全球购物
草莓网官网:StrawberryNET
2019/08/21 全球购物
科长竞争上岗演讲稿
2014/05/12 职场文书
中职毕业生自我鉴定
2014/09/13 职场文书
学生旷课检讨书500字
2014/10/28 职场文书
2015年挂职锻炼工作总结
2014/12/12 职场文书
投标承诺函范文
2015/01/21 职场文书
狼牙山五壮士观后感
2015/06/09 职场文书
2016年大学生党员公开承诺书
2016/03/24 职场文书
Nginx隐藏式跳转(浏览器URL跳转后保持不变)
2022/04/07 Servers
Android开发实现极为简单的QQ登录页面
2022/04/24 Java/Android