pytorch模型预测结果与ndarray互转方式
作者:fly_Xiaoma 时间:2023-12-06 02:35:11
预测结果转为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)
来源:https://blog.csdn.net/weixin_38664232/article/details/103550448
标签:pytorch,模型,预测结果,ndarray
0
投稿
猜你喜欢
Python设计模式中的创建型工厂模式
2023-05-09 17:47:02
Python 网页解析HTMLParse的实例详解
2022-08-29 06:27:25
js+html5通过canvas指定开始和结束点绘制线条的方法
2024-05-02 17:20:44
SQLSERVER分布式事务使用实例
2024-01-28 03:57:28
一文深入了解Python中的继承知识点
2023-01-16 19:57:16
数据库访问性能优化
2024-01-21 18:24:47
numpy的sum函数的axis和keepdim参数详解
2021-08-02 02:48:46
基于MATLAB和Python实现MFCC特征参数提取
2023-01-29 07:37:53
二级联动下拉菜单javascript源码
2010-03-16 12:32:00
SQL Server 2005数据库中表的递归查询
2009-01-08 16:08:00
python list删除元素时要注意的坑点分享
2021-07-15 16:02:22
ASP.net如何连接SQL SERVER 2012数据库
2024-01-24 12:34:53
IE10增强对HTML5和CSS3的支持
2011-09-16 20:16:28
Python 生成一个从0到n个数字的列表4种方法小结
2023-06-10 19:14:59
MySQL内外连接的具体使用
2024-01-17 13:52:49
Python3网络爬虫开发实战之极验滑动验证码的识别
2022-10-02 22:06:14
利用canvas实现的加载动画效果实例代码
2023-06-29 22:49:51
Python中的np.argmin()和np.argmax()函数用法
2021-09-10 16:00:51
python使用pil进行图像处理(等比例压缩、裁剪)实例代码
2022-12-22 16:50:09
简洁的是最好的吗?
2010-06-24 21:46:00