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小技巧之批量抓取美女图片
Jun 06 Python
快速入门python学习笔记
Dec 06 Python
Python多层装饰器用法实例分析
Feb 09 Python
Python向MySQL批量插数据的实例讲解
Mar 31 Python
Python实现正弦信号的时域波形和频谱图示例【基于matplotlib】
May 04 Python
tensorflow实现逻辑回归模型
Sep 08 Python
python 读取dicom文件,生成info.txt和raw文件的方法
Jan 24 Python
python3 tkinter实现点击一个按钮跳出另一个窗口的方法
Jun 13 Python
Django model 中设置联合约束和联合索引的方法
Aug 06 Python
python flask搭建web应用教程
Nov 19 Python
django实现将后台model对象转换成json对象并传递给前端jquery
Mar 16 Python
Python tornado上传文件的功能
Mar 26 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中文件上传的安全问题
2006/10/09 PHP
Php Cookie的一个使用注意点
2008/11/08 PHP
php smarty模版引擎中变量操作符及使用方法
2009/12/11 PHP
PHP中PDO基础教程 入门级
2011/09/04 PHP
php随机显示指定文件夹下图片的方法
2015/07/13 PHP
从ThinkPHP3.2.3过渡到ThinkPHP5.0学习笔记图文详解
2019/04/03 PHP
GWT中复制到剪贴板 js+flash实现复制 兼容性比较好
2010/03/07 Javascript
Uglifyjs(JS代码优化工具)入门 安装使用
2020/04/13 Javascript
js点击按钮实现水波纹效果代码(CSS3和Canves)
2016/09/15 Javascript
JQuery实现DIV其他动画效果的简单实例
2016/09/18 Javascript
JS判断输入的字符串是否是数字的方法(正则表达式)
2016/11/29 Javascript
xmlplus组件设计系列之按钮(2)
2017/04/26 Javascript
js实现放大镜特效
2017/05/18 Javascript
在Js页面通过POST传递参数跳转到新页面详解
2017/08/25 Javascript
JS实现的数组去除重复数据算法小结
2017/11/17 Javascript
vue.js中引入vuex储存接口数据及调用的详细流程
2017/12/14 Javascript
vue router动态路由设置参数可选问题
2019/08/21 Javascript
[01:33]真香警告!DOTA2勇士令状不朽珍藏Ⅱ饰品欣赏
2018/06/26 DOTA
python自动格式化json文件的方法
2015/03/11 Python
python和shell监控linux服务器的详细代码
2018/06/22 Python
Appium+Python自动化测试之运行App程序示例
2019/01/23 Python
python实现简单俄罗斯方块
2020/03/13 Python
python查看矩阵的行列号以及维数方式
2020/05/22 Python
Python Tkinter实例——模拟掷骰子
2020/10/24 Python
python脚本使用阿里云slb对恶意攻击进行封堵的实现
2021/02/04 Python
北美女性服装零售连锁店:maurices
2019/06/12 全球购物
面向对象编程是如何提高软件开发水平的
2014/05/06 面试题
法律六进活动方案
2014/03/13 职场文书
法律进学校实施方案
2014/03/15 职场文书
网络优化专员求职信
2014/05/04 职场文书
2014广电局实施党的群众路线教育实践活动方案思想汇报
2014/09/22 职场文书
合伙购房协议样本
2014/10/06 职场文书
党员教师群众路线思想汇报范文
2014/10/28 职场文书
国际贸易实务实训报告
2014/11/05 职场文书
2015年保卫科工作总结
2015/05/14 职场文书
redis实现的四种常见限流策略
2021/06/18 Redis