這篇文章主要介紹了使用pytorch怎么對tensor與numpy數(shù)組進行轉換，此處通過實例代碼給大家介紹的非常詳細，對大家的學習或工作具有一定的參考價值，需要的朋友可以參考下：

創(chuàng)新互聯(lián)網站建設服務商，為中小企業(yè)提供網站建設、網站制作服務，網站設計，成都網站托管等一站式綜合服務型公司，專業(yè)打造企業(yè)形象網站，讓您在眾多競爭對手中脫穎而出創(chuàng)新互聯(lián)。

a = torch.ones(2,2)
b = a.numpy()
c=np.array(a) #也可以轉numpy數(shù)組
print(type(a))
print(type(b))
print(a)
print(b)

輸出為：

<class ‘torch.Tensor'>
<class ‘numpy.ndarray'>
tensor([[1., 1.],
[1., 1.]])
[[1. 1.]
[1. 1.]]

numpy轉tensor：

import torch
import numpy as np

a = np.ones(5)
b = torch.from_numpy(a)
c=torch.Tensor(a) #也可以轉pytorch Tensor
print(type(a))
print(type(b))
print(a)
print(b)

輸出為：

<class ‘numpy.ndarray'>
<class ‘torch.Tensor'>
[1. 1. 1. 1. 1.]
tensor([1., 1., 1., 1., 1.], dtype=torch.float64)

可見pytorch的tensor對象與numpy數(shù)組是可以相互轉換的，且numpy數(shù)組的默認類型是double

到此這篇關于使用pytorch怎么對tensor與numpy數(shù)組進行轉換的文章就介紹到這了,更多相關使用pytorch怎么對tensor與numpy數(shù)組進行轉換的內容請搜索創(chuàng)新互聯(lián)成都網站設計公司以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持創(chuàng)新互聯(lián)成都網站設計公司！

另外有需要云服務器可以了解下創(chuàng)新互聯(lián)scvps.cn，海內外云服務器15元起步，三天無理由+7*72小時售后在線，公司持有idc許可證，提供“云服務器、裸金屬服務器、高防服務器、香港服務器、美國服務器、虛擬主機、免備案服務器”等云主機租用服務以及企業(yè)上云的綜合解決方案，具有“安全穩(wěn)定、簡單易用、服務可用性高、性價比高”等特點與優(yōu)勢，專為企業(yè)上云打造定制，能夠滿足用戶豐富、多元化的應用場景需求。

分享名稱：使用pytorch怎么對tensor與numpy數(shù)組進行轉換-創(chuàng)新互聯(lián)
文章轉載：http://jinyejixie.com/article6/djscig.html

成都網站建設公司_創(chuàng)新互聯(lián)，為您提供網站內鏈、云服務器、網站改版、電子商務、Google、網站導航

廣告

聲明：本網站發(fā)布的內容（圖片、視頻和文字）以用戶投稿、用戶轉載內容為主，如果涉及侵權請盡快告知，我們將會在第一時間刪除。文章觀點不代表本網站立場，如需處理請聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內容未經允許不得轉載，或轉載時需注明來源： 創(chuàng)新互聯(lián)