Pyorch之numpy與torch之間相互轉(zhuǎn)換方式
更新時間：2019年12月31日 09:26:28 作者：嘖嘖嘖biubiu
今天小編就為大家分享一篇Pyorch之numpy與torch之間相互轉(zhuǎn)換方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
numpy中的ndarray轉(zhuǎn)化成pytorch中的tensor : torch.from_numpy()
pytorch中的tensor轉(zhuǎn)化成numpy中的ndarray : numpy()
代碼
import numpy as np
import torch
np_arr = np.array([1,2,3,4])
tor_arr=torch.from_numpy(np_arr)
tor2numpy=tor_arr.numpy()
print('\nnumpy\n',np_arr,'\ntorch\n',tor_arr,'\nnumpy\n',tor2numpy)
輸出
numpy [1 2 3 4] torch tensor([1, 2, 3, 4], dtype=torch.int32) numpy [1 2 3 4]
以上這篇Pyorch之numpy與torch之間相互轉(zhuǎn)換方式就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
Pandas實現(xiàn)復(fù)制dataframe中的每一行
這篇文章主要介紹了Pandas實現(xiàn)復(fù)制dataframe中的每一行方式,2024-02-02
這篇文章主要介紹了Python基于execjs運行js過程解析,文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-11-11
這篇文章主要介紹了Python的加密模塊md5、sha、crypt使用實例,本文給出了MD5和crypt模塊的代碼實例,需要的朋友可以參考下2014-09-09
Python?數(shù)據(jù)可視化神器Pyecharts繪制圖像練習(xí)
這篇文章主要介紹了Python?數(shù)據(jù)可視化神器Pyecharts繪制圖像練習(xí)，繪制的圖形有柱狀圖、餅狀圖、箱型圖、折線圖、雷達圖等多種圖像，需要的小伙伴可以參考一下2022-02-02
Python模擬鼠標點擊實現(xiàn)方法(將通過實例自動化模擬在360瀏覽器中自動搜索python)
這篇文章主要介紹了Python模擬鼠標點擊實現(xiàn)方法(將通過實例自動化模擬在360瀏覽器中自動搜索python),需要的朋友可以參考下2017-08-08
python用pickle模塊實現(xiàn)“增刪改查”的簡易功能
本篇文章主要介紹了python用pickle模塊實現(xiàn)“增刪改查”的簡易功能，具有一定的參考價值，感興趣的小伙伴們可以參考一下。2017-06-06
timeit 模塊是 Python 標準庫中的模塊，無需安裝，直接導(dǎo)入就可以使用。這篇文章主要介紹了Python timeit模塊的使用,需要的朋友可以參考下2020-01-01
解決Keras中循環(huán)使用K.ctc_decode內(nèi)存不釋放的問題
這篇文章主要介紹了解決Keras中循環(huán)使用K.ctc_decode內(nèi)存不釋放的問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-06-06
對Python 窗體(tkinter)文本編輯器(Text)詳解
今天小編就為大家分享一篇對Python 窗體(tkinter)文本編輯器(Text)詳解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-10-10
在本篇內(nèi)容里小編給各位分享的是一篇關(guān)于python中的插入排序的簡單用法，有興趣的朋友們可以參考學(xué)習(xí)下。2021-01-01