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设计】。

相关文章

Python3遍历目录树实现方法

本文实例讲述了Python3遍历目录树的方法。分享给大家供大家参考。具体实现方法如下: import os, fnmatch # 检查一个目录,后者某个包含子目录的目录树,并根据某种...

Python简单生成随机数的方法示例

本文实例讲述了Python简单生成随机数的方法。分享给大家供大家参考,具体如下: 主要知识点: 随机整数: random.randint(a,b):返回随机整数x,a<=x<...

基于Python的XSS测试工具XSStrike使用方法

基于Python的XSS测试工具XSStrike使用方法

简介 XSStrike 是一款用于探测并利用XSS漏洞的脚本 XSStrike目前所提供的产品特性: 对参数进行模糊测试之后构建合适的payload 使用payload对参数进行穷举匹配...

Python IDE Pycharm中的快捷键列表用法

Python IDE Pycharm中的快捷键列表用法

常用快捷键 全部快捷键 1、编辑(Editing) 2、查找/替换(Search/Replace) 3、运行(Running) 4、调试(Debugging) 5、导航(N...

理解python正则表达式

在python中,对正则表达式的支持是通过re模块来支持的。使用re的步骤是先把表达式字符串编译成pattern实例,然后在使用pattern去匹配文本获取结果。 其实也有另外一种方式,...