pytorch模型预测结果与ndarray互转方式
更新时间：2020年01月15日 10:18:28 作者：fly_Xiaoma
今天小编就为大家分享一篇pytorch模型预测结果与ndarray互转方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
预测结果转为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互转方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
本文主要介绍了全网非常详细的pytest配置文件，pytest的主配置文件，可以改变pytest的默认行为，有很多可配置的选项，感兴趣的可以了解一下2022-07-07
这篇文章主要介绍了Python中使用HTMLParser解析html实例,本文直接给出使用示例,并总结出HTMLParser含有的方法分为两类,一类是需要显式调用的,而另一类不需显示调用,需要的朋友可以参考下2015-02-02
pytest自动化测试fixture的作用域实例化顺序及可用性
这篇文章主要介绍了pytest自动化测试中fixture的作用域、实例化顺序及可用性的详解示例有需要的朋友可以借鉴参考下，希望能够有所帮助2021-10-10
这篇文章主要介绍了python查看包版本、更新单个包、卸载单个包,本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2022-12-12
今天小编就为大家分享一篇python+selenium 鼠标事件操作方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-08-08
通过本文的介绍,相信你已经对PyQt5中的QPushButton控件有了深入的了解,从基础介绍到常用属性和方法,再到应用场景和样式定制,本文为你提供了全面的指南,感兴趣的朋友跟随小编一起看看吧2024-08-08
在本篇内容里小编给大家整理了关于python3中calendar返回某一时间点实例讲解内容，有兴趣的朋友们可以参考学习下。2020-11-11
今天小编就为大家分享一篇获取python的list中含有重复值的index方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-06-06
Python办公⾃动化是利⽤Python编程语⾔来创建脚本和程序,以简化、加速和⾃动化⽇常办公任务和⼯作流程的过程,本文主要来介绍一下如何利用Python进行CSV文件的运用和管理,感兴趣的可以了解下2023-12-12
pandas是一个十分强大的数据处理工具,最近需要处理数据并输入到excel,简单列举它的用法,这篇文章主要给大家介绍了关于python使用pandas读写excel文件的相关资料,文中通过实例代码介绍的非常详细,需要的朋友可以参考下2022-08-08
最新评论