pytorch模型預(yù)測結(jié)果與ndarray互轉(zhuǎn)方式
更新時間：2020年01月15日 10:18:28 作者：fly_Xiaoma
今天小編就為大家分享一篇pytorch模型預(yù)測結(jié)果與ndarray互轉(zhuǎn)方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
預(yù)測結(jié)果轉(zhuǎn)為numpy：
logits=model(feature) #如果模型是跑在GPU上 result=logits.data.cpu().numpy() / logits.cpu().numpy() #如果模型跑在cpu上 result=logits.data.numpy() / logits.numpy()
將矩陣轉(zhuǎn)為tensor：
np_arr = np.array([1,2,3,4]) tensor=torch.from_numpy(np_arr)
以上這篇pytorch模型預(yù)測結(jié)果與ndarray互轉(zhuǎn)方式就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
本文主要介紹了全網(wǎng)非常詳細(xì)的pytest配置文件，pytest的主配置文件，可以改變pytest的默認(rèn)行為，有很多可配置的選項(xiàng)，感興趣的可以了解一下2022-07-07
Python中使用HTMLParser解析html實(shí)例
這篇文章主要介紹了Python中使用HTMLParser解析html實(shí)例,本文直接給出使用示例,并總結(jié)出HTMLParser含有的方法分為兩類,一類是需要顯式調(diào)用的,而另一類不需顯示調(diào)用,需要的朋友可以參考下2015-02-02
這篇文章主要介紹了python查看包版本、更新單個包、卸載單個包,本文通過實(shí)例代碼給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下
2022-12-12 
python+selenium 鼠標(biāo)事件操作方法
今天小編就為大家分享一篇python+selenium 鼠標(biāo)事件操作方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2019-08-08 
一文詳細(xì)介紹PyQt5 QPushButton() 的作用
通過本文的介紹,相信你已經(jīng)對PyQt5中的QPushButton控件有了深入的了解,從基礎(chǔ)介紹到常用屬性和方法,再到應(yīng)用場景和樣式定制,本文為你提供了全面的指南,感興趣的朋友跟隨小編一起看看吧
2024-08-08 
python3中calendar返回某一時間點(diǎn)實(shí)例講解
在本篇內(nèi)容里小編給大家整理了關(guān)于python3中calendar返回某一時間點(diǎn)實(shí)例講解內(nèi)容，有興趣的朋友們可以參考學(xué)習(xí)下。
2020-11-11 
獲取python的list中含有重復(fù)值的index方法
今天小編就為大家分享一篇獲取python的list中含有重復(fù)值的index方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2018-06-06 
Python辦公?動化是利?Python編程語?來創(chuàng)建腳本和程序,以簡化、加速和?動化?常辦公任務(wù)和?作流程的過程,本文主要來介紹一下如何利用Python進(jìn)行CSV文件的運(yùn)用和管理,感興趣的可以了解下
2023-12-12 
python使用pandas讀寫excel文件的方法實(shí)例
pandas是一個十分強(qiáng)大的數(shù)據(jù)處理工具,最近需要處理數(shù)據(jù)并輸入到excel,簡單列舉它的用法,這篇文章主要給大家介紹了關(guān)于python使用pandas讀寫excel文件的相關(guān)資料,文中通過實(shí)例代碼介紹的非常詳細(xì),需要的朋友可以參考下
2022-08-08