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)文章
這篇文章主要介紹了如何利用python畫畫，幫助大家更好的理解和使用python的turtle庫，感興趣的朋友可以了解下2020-12-12
這篇文章主要介紹了Python生成器generator用法,結(jié)合實例形式分析了Python生成器generator常見操作技巧與相關(guān)注意事項,需要的朋友可以參考下2018-08-08
python自動打開瀏覽器下載zip并提取內(nèi)容寫入excel
這篇文章主要給大家介紹了關(guān)于python自動打開瀏覽器下載zip并提取內(nèi)容寫入excel的相關(guān)資料，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-01-01
本文主要介紹了python加速器numba使用詳解，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-02-02
Python編程快速上手——Excel表格創(chuàng)建乘法表案例分析
這篇文章主要介紹了Python Excel表格創(chuàng)建乘法表,結(jié)合具體實例形式分析了Python接受cmd命令操作Excel文件創(chuàng)建乘法表相關(guān)實現(xiàn)技巧,需要的朋友可以參考下2020-02-02
176萬！GPT-4發(fā)布詳解如何查看OpenAI的下載量
這篇文章主要為大家介紹了GPT-4發(fā)布如何查看OpenAI的下載量詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2023-03-03
這篇文章主要介紹了python實現(xiàn)二叉排序樹，2022-01-01
python實現(xiàn)將漢字轉(zhuǎn)換成漢語拼音的庫
這篇文章主要介紹了python實現(xiàn)將漢字轉(zhuǎn)換成漢語拼音的庫,涉及Python調(diào)用word.data字段實現(xiàn)將漢字轉(zhuǎn)換成拼音的功能,非常具有實用價值,需要的朋友可以參考下2015-05-05
在Python中,引用不確定的函數(shù)通常意味著我們可能在運行時才知道要調(diào)用哪個函數(shù),或者我們可能想根據(jù)某些條件動態(tài)地選擇不同的函數(shù)來執(zhí)行,下面給大家分享Python?如何引用不確定的函數(shù),感興趣的朋友跟隨小編一起看看吧2024-07-07
Python中循環(huán)后使用list.append()數(shù)據(jù)被覆蓋問題的解決
這篇文章主要給大家介紹了關(guān)于Python中循環(huán)后使用list.append()數(shù)據(jù)被覆蓋問題的解決方法，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2018-07-07