pytorch模型预测结果与ndarray互转方式

yipeiwu_com6年前Python基础

预测结果转为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装饰器计算函数运行时间的实例

装饰器在python里面有很重要的作用, 如果能够熟练使用,将会大大的提高工作效率 今天就来见识一下 python 装饰器,到底是怎么工作的。 本文主要是利用python装饰器计算函数运...

Python画图学习入门教程

本文实例讲述了Python画图的基本方法。分享给大家供大家参考,具体如下: Python:使用matplotlib绘制图表 python绘制图表的方法,有个强大的类库matplotlib...

对python中的xlsxwriter库简单分析

一、xlsxwriter 基本用法,创建 xlsx 文件并添加数据 官方文档:http://xlsxwriter.readthedocs.org/ xlsxwriter 可以操作 xls...

python 生成器协程运算实例

一、yield运行方式 我们定义一个如下的生成器: def put_on(name): print("Hi {}, 货物来了,准备搬到仓库!".format(name)) wh...

关于django 数据库迁移(migrate)应该知道的一些事

命令 首先数据库迁移的两大命令: python manage.py makemigrations & python manage.py migrate 前者是将model层转为...