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 相关文章推荐
linux系统使用python获取内存使用信息脚本分享
Jan 15 Python
python解析html开发库pyquery使用方法
Feb 07 Python
Python socket.error: [Errno 98] Address already in use的原因和解决方法
Aug 25 Python
Python中for循环控制语句用法实例
Jun 02 Python
Windows下python2.7.8安装图文教程
May 26 Python
python中如何使用正则表达式的非贪婪模式示例
Oct 09 Python
Python实现翻转数组功能示例
Jan 12 Python
python3.6使用pickle序列化class的方法
Oct 22 Python
关于阿里云oss获取sts凭证 app直传 python的实例
Aug 20 Python
PyTorch里面的torch.nn.Parameter()详解
Jan 03 Python
pytorch 修改预训练model实例
Jan 18 Python
Python小白不正确的使用类变量实例
May 29 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 配置open_basedir 让各虚拟站点独立运行
2009/11/12 PHP
php 缩略图实现函数代码
2011/06/23 PHP
一个PHP针对数字的加密解密类
2014/03/20 PHP
浅谈PHP中try{}catch{}的使用方法
2016/12/09 PHP
JavaScript的parseInt 进制问题
2009/05/07 Javascript
DIY jquery plugin - tabs标签切换实现代码
2010/12/11 Javascript
下拉菜单点击实现连接跳转功能的js代码
2013/05/19 Javascript
JS未跨域操作iframe里的DOM
2016/06/01 Javascript
JavaScript动态检验密码强度的实现方法
2016/11/09 Javascript
Javascript 引擎工作机制详解
2016/11/30 Javascript
js 判断登录界面的账号密码是否为空
2017/02/08 Javascript
JavaScript算法教程之sku(库存量单位)详解
2017/06/29 Javascript
详解webpack+gulp实现自动构建部署
2017/06/29 Javascript
node.js通过axios实现网络请求的方法
2018/03/05 Javascript
vue中引入第三方字体文件的方法示例
2018/12/17 Javascript
原来JS还可以这样拆箱转换详解
2019/02/01 Javascript
Vue退出登录时清空缓存的实现
2019/11/12 Javascript
写给新手同学的vuex快速上手指北小结
2020/04/14 Javascript
如何在JS文件中获取Vue组件
2020/09/16 Javascript
node.js 基于 STMP 协议和 EWS 协议发送邮件
2021/02/14 Javascript
python杀死一个线程的方法
2015/09/06 Python
python实现TF-IDF算法解析
2018/01/02 Python
在Django中预防CSRF攻击的操作
2020/03/13 Python
python模拟哔哩哔哩滑块登入验证的实现
2020/04/24 Python
Selenium元素定位的30种方式(史上最全)
2020/05/11 Python
keras 自定义loss层+接受输入实例
2020/06/28 Python
与C++相比,Java中的数组有什么不同
2014/03/25 面试题
Java中的类包括什么内容?设计时要注意哪些方面
2012/05/23 面试题
国际贸易专业求职信
2014/06/04 职场文书
2013年最新自荐信范文
2014/06/23 职场文书
禁毒宣传活动总结
2014/08/26 职场文书
民间借贷协议书范本
2014/10/01 职场文书
法定代表人免职证明
2015/06/24 职场文书
承诺书怎么写 ?
2019/04/16 职场文书
Go语言基础函数基本用法及示例详解
2021/11/17 Golang
MySQL实现字段分割一行转多行的示例代码
2022/07/07 MySQL